Use numpy for cumprod

pymc-labs · Jan 29, 2024 · a208794 · a208794
1 parent 56240b7
commit a208794
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/mmm/test_transformers.py b/tests/mmm/test_transformers.py
@@ -212,7 +212,7 @@ def test_weibull_cdf_adsotck(self, x, lam, k, l_max):
 
         assert np.all(np.isfinite(y))
         w = 1 - sp.stats.weibull_min.cdf(np.arange(l_max) + 1, c=k, scale=lam)
-        w = sp.cumprod(np.concatenate([[1], w]))
+        w = np.cumprod(np.concatenate([[1], w]))
         sp_y = batched_convolution(x, w).eval()
         np.testing.assert_almost_equal(y, sp_y)