fix make_evaluate_...

Author: penelopeysm

src/mcmc/particle_mcmc.jl

@@ -18,15 +18,16 @@ function TracedModel(
     varinfo::AbstractVarInfo,
     rng::Random.AbstractRNG,
 )
-    context = DynamicPPL.SamplingContext(rng, sampler, DefaultContext())
-    args, kwargs = DynamicPPL.make_evaluate_args_and_kwargs(model, varinfo, context)
+    spl_context = DynamicPPL.SamplingContext(rng, sampler, model.context)
+    spl_model = DynamicPPL.contextualize(model, spl_context)
+    args, kwargs = DynamicPPL.make_evaluate_args_and_kwargs(spl_model, varinfo)
     if kwargs !== nothing && !isempty(kwargs)
         error(
             "Sampling with `$(sampler.alg)` does not support models with keyword arguments. See issue #2007 for more details.",
         )
     end
     return TracedModel{AbstractSampler,AbstractVarInfo,Model,Tuple}(
-        model, sampler, varinfo, (model.f, args...)
+        spl_model, sampler, varinfo, (spl_model.f, args...)
     )
 end