fix

Author: penelopeysm

src/debug_utils.jl

@@ -438,24 +438,26 @@ function check_model_and_trace(
     rng::Random.AbstractRNG,
     model::Model;
     varinfo=VarInfo(),
-    context=SamplingContext(rng),
     error_on_failure=false,
     kwargs...,
 )
     # Execute the model with the debug context.
     debug_context = DebugContext(
-        model, context; error_on_failure=error_on_failure, kwargs...
+        model,
+        SamplingContext(rng, model.context);
+        error_on_failure=error_on_failure,
+        kwargs...,
     )
+    debug_model = DynamicPPL.contextualize(model, debug_context)
 
     # Perform checks before evaluating the model.
-    issuccess = check_model_pre_evaluation(debug_context, model)
+    issuccess = check_model_pre_evaluation(debug_context, debug_model)
 
     # Force single-threaded execution.
-    debug_model = DynamicPPL.contextualize(model, debug_context)
     DynamicPPL.evaluate_threadunsafe!!(debug_model, varinfo)
 
     # Perform checks after evaluating the model.
-    issuccess &= check_model_post_evaluation(debug_context, model)
+    issuccess &= check_model_post_evaluation(debug_context, debug_model)
 
     if !issuccess && error_on_failure
         error("model check failed")