From a208794bc28e21b06067b42560edb709891fea49 Mon Sep 17 00:00:00 2001 From: Abdalaziz Rashid Date: Mon, 29 Jan 2024 13:59:31 +0000 Subject: [PATCH] Use numpy for cumprod --- tests/mmm/test_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/mmm/test_transformers.py b/tests/mmm/test_transformers.py index fb98c2d51..61595aa5d 100644 --- 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)