From ee23a8cfff86c5047a917f5e8371f3fd2f719c30 Mon Sep 17 00:00:00 2001
From: Michal Raczycki <michalr265@gmail.com>
Date: Sun, 20 Aug 2023 14:23:20 +0200
Subject: [PATCH] adjusting plotting fit

---
 tests/mmm/test_plotting.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/mmm/test_plotting.py b/tests/mmm/test_plotting.py
index 4d4a8cf6..b94961e6 100644
--- 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