changed N to 4000

pymc-labs · Jul 14, 2023 · 3b7f58e · 3b7f58e
1 parent 4d5b95f
commit 3b7f58e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pymc_marketing/clv/models/pareto_nbd.py b/pymc_marketing/clv/models/pareto_nbd.py
@@ -633,7 +633,7 @@ def _population_distributions(
 
             # This is the shape if using fit_method="map"
             if self.fit_result.dims == {"chain": 1, "draw": 1}:
-                shape_kwargs = {"shape": 10000}
+                shape_kwargs = {"shape": 4000}
             else:
                 shape_kwargs = {}
 

diff --git a/tests/clv/models/test_pareto_nbd.py b/tests/clv/models/test_pareto_nbd.py
@@ -361,7 +361,7 @@ def test_dropout_purchase_distributions(self) -> None:
         assert isinstance(customer_dropout, xarray.DataArray)
         assert isinstance(customer_purchase_rate, xarray.DataArray)
 
-        N = 10000
+        N = 4000
         lam = pm.Gamma.dist(self.r_true, self.alpha_true, size=N)
         mu = pm.Gamma.dist(self.s_true, self.beta_true, size=N)
         rtol = 0.05