adjusting plotting fit

pymc-labs · Aug 20, 2023 · ee23a8c · ee23a8c
1 parent 381daee
commit ee23a8c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/mmm/test_plotting.py b/tests/mmm/test_plotting.py
@@ -75,8 +75,11 @@ class ToyMMM(BaseDelayedSaturatedMMM, MaxAbsScaleTarget):
         mmm.fit(
             X=toy_X,
             y=toy_y,
-            prior_predictive=True,
-            posterior_predictive=True,
+            sample_kwargs={
+                "prior_kwargs": {"samples": 100},
+                "prior_predictive": True,
+                "posterior_predictive": True,
+            },
         )
         mmm._prior_predictive = mmm.prior_predictive
         mmm._fit_result = mmm.fit_result