Add suggestions from code review

Co-authored-by: Tor Erlend Fjelde <[email protected]>
TuringLang · Nov 29, 2024 · 7d6a983 · 7d6a983
1 parent 38ac128
commit 7d6a983
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mcmc/gibbs.jl b/src/mcmc/gibbs.jl
@@ -509,7 +509,7 @@ function setparams_varinfo!!(
     # update its logprob. To do this, we have to call evaluate!! with the sampler, rather
     # than just a context, because ESS is peculiar in how it uses LikelihoodContext for
     # some variables and DefaultContext for others.
-    return last(DynamicPPL.evaluate!!(model, params, sampler))
+    return last(DynamicPPL.evaluate!!(model, params, SamplingContext(sampler)))
 end
 
 function setparams_varinfo!!(