fix to be compatible with numpy > 1.16

alan-turing-institute · Dec 13, 2024 · cfdc10e · cfdc10e
1 parent 0c45de0
commit cfdc10e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -156,7 +156,7 @@ def test_prediction_shapes_lowlevel(self, n_target_sets):
             ):
                 task = tl("2020-01-01", context_sampling, target_sampling)
 
-                n_targets = np.product(expected_obs_shape)
+                n_targets = np.prod(expected_obs_shape)
 
                 # Tensors
                 mean = model.mean(task)
@@ -192,7 +192,7 @@ def test_prediction_shapes_lowlevel(self, n_target_sets):
                     )
 
                 if likelihood in ["cnp", "gnp"]:
-                    n_target_dims = np.product(tl.target_dims)
+                    n_target_dims = np.prod(tl.target_dims)
                     assert_shape(
                         model.covariance(task),
                         (