fix to be compatible with numpy > 1.16

davidwilby · davidwilby · commit cfdc10ec4970 · 2024-12-13T16:36:52.000Z
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),
                         (

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def test_prediction_shapes_lowlevel(self, n_target_sets):`
`156`	`156`	`):`
`157`	`157`	`task = tl("2020-01-01", context_sampling, target_sampling)`
`158`	`158`
`159`		`- n_targets = np.product(expected_obs_shape)`
	`159`	`+ n_targets = np.prod(expected_obs_shape)`
`160`	`160`
`161`	`161`	`# Tensors`
`162`	`162`	`mean = model.mean(task)`
`@@ -192,7 +192,7 @@ def test_prediction_shapes_lowlevel(self, n_target_sets):`
`192`	`192`	`)`
`193`	`193`
`194`	`194`	`if likelihood in ["cnp", "gnp"]:`
`195`		`- n_target_dims = np.product(tl.target_dims)`
	`195`	`+ n_target_dims = np.prod(tl.target_dims)`
`196`	`196`	`assert_shape(`
`197`	`197`	`model.covariance(task),`
`198`	`198`	`(`