Fix #129

Author: devmotion

src/context_implementations.jl

@@ -177,7 +177,7 @@ function dot_tilde(
     else
         vns, dist = get_vns_and_dist(right, left, vn)
     end
-    return _dot_tilde(rng, sampler, NoDist(dist), left, vns, vi)
+    return _dot_tilde(rng, sampler, NoDist.(dist), left, vns, vi)
 end
 function dot_tilde(rng, ctx::MiniBatchContext, sampler, right, left, vn::VarName, inds, vi)
     return dot_tilde(rng, ctx.ctx, sampler, right, left, vn, inds, vi)

test/context_implementations.jl

@@ -0,0 +1,15 @@
+@testset "context_implementation.jl" begin
+    # https://github.com/TuringLang/DynamicPPL.jl/issues/129
+    @testset "#129" begin
+        @model function test(x)
+            μ ~ MvNormal(fill(0, 2), 2.0)
+            z = Vector{Int}(undef, length(x))
+            z .~ Categorical.(fill([0.5, 0.5], length(x)))
+            for i in 1:length(x)
+                x[i] ~ Normal(μ[z[i]], 0.1)
+            end
+        end
+
+        test([1, 1, -1])(VarInfo(), SampleFromPrior(), LikelihoodContext())
+    end
+end

test/runtests.jl

@@ -25,6 +25,7 @@ include("test_util.jl")
     include("prob_macro.jl")
     include("independence.jl")
     include("distribution_wrappers.jl")
+    include("context_implementations.jl")
 
     include("threadsafe.jl")