Update ext/DynamicPPLTestExt/contexts.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Nov 27, 2024 · 298ec36 · 298ec36
1 parent b93e581
commit 298ec36
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/ext/DynamicPPLTestExt/contexts.jl b/ext/DynamicPPLTestExt/contexts.jl
@@ -29,9 +29,15 @@ function DynamicPPL.TestUtils.TestLogModifyingChildContext(
     )
 end
 
-DynamicPPL.NodeTrait(::DynamicPPL.TestUtils.TestLogModifyingChildContext) = DynamicPPL.IsParent()
-DynamicPPL.childcontext(context::DynamicPPL.TestUtils.TestLogModifyingChildContext) = context.context
-function DynamicPPL.setchildcontext(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, child)
+function DynamicPPL.NodeTrait(::DynamicPPL.TestUtils.TestLogModifyingChildContext)
+    return DynamicPPL.IsParent()
+end
+function DynamicPPL.childcontext(context::DynamicPPL.TestUtils.TestLogModifyingChildContext)
+    return context.context
+end
+function DynamicPPL.setchildcontext(
+    context::DynamicPPL.TestUtils.TestLogModifyingChildContext, child
+)
     return DynamicPPL.TestUtils.TestLogModifyingChildContext(context.mod, child)
 end
 function DynamicPPL.tilde_assume(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, vn, vi)