use evaluate!! instead

TuringLang · May 15, 2024 · 09cae5b · 09cae5b
1 parent 1a67b5f
commit 09cae5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sampler.jl b/src/sampler.jl
@@ -67,7 +67,7 @@ function AbstractMCMC.step(
         vi = VarInfo()
         model(rng, vi, sampler)
     else
-        vi = model(rng, SimpleVarInfo{Float64}(OrderedDict()), sampler)
+        vi = last(evaluate!!(model, rng, SimpleVarInfo{Float64}(OrderedDict()), sampler))
     end
     return vi, nothing
 end