Update test_friedman_fit.py

np.alltrue to np.all for numpy 2.0 compatibility
lanl · Sep 20, 2024 · 30b7eee · 30b7eee
1 parent 6e6081b
commit 30b7eee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_friedman_fit.py b/tests/test_friedman_fit.py
@@ -115,5 +115,5 @@ def test_friedman_fit():
     # Test that the true values of theta are in the 95% credible interval.
     lower = np.quantile(theta_posterior, .025, 0)
     upper = np.quantile(theta_posterior, .975, 0)
-    assert np.alltrue(lower < param_truth['theta'])
-    assert np.alltrue(param_truth['theta'] < upper)
+    assert np.all(lower < param_truth['theta'])
+    assert np.all(param_truth['theta'] < upper)