Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Nov 16, 2023 · afe5454 · afe5454
1 parent e6a06a8
commit afe5454
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/varinfo.jl b/test/varinfo.jl
@@ -598,11 +598,11 @@ DynamicPPL.getspace(::DynamicPPL.Sampler{MySAlg}) = (:s,)
         @testset "different models" begin
             @model function demo_merge_different_y()
                 x ~ Uniform()
-                y ~ Normal()
+                return y ~ Normal()
             end
             @model function demo_merge_different_z()
                 x ~ Normal()
-                z ~ Normal()
+                return z ~ Normal()
             end
             model_left = demo_merge_different_y()
             model_right = demo_merge_different_z()