Use model argument in Evaluator when getting model context

exoplanet-dev · Nov 3, 2023 · 3cc0812 · 3cc0812
1 parent 91b8026
commit 3cc0812
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pymc_ext/utils.py b/src/pymc_ext/utils.py
@@ -18,7 +18,7 @@ def __init__(self, outs, model=None, **kwargs):
         # In PyMC >= 5, model context is required to replace rvs with values
         # https://github.com/pymc-devs/pymc/pull/6281
         # https://www.pymc.io/projects/docs/en/stable/learn/core_notebooks/pymc_pytensor.html#pymc
-        model = pm.modelcontext(None)
+        model = pm.modelcontext(model)
         if isinstance(outs, (tuple, list)):
             self.out_values = model.replace_rvs_by_values(outs)
         else: