conftest.py: running tests with numpy 2.0 (#100)

np.product was removed in numpy 2.0, and only np.prod remains. This commit allows tests to pass under numpy 2.0
mmuckley · Nov 22, 2024 · 7e23776 · 7e23776
1 parent 7039bc1
commit 7e23776
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -26,7 +26,7 @@ def test_params():
 
 
 def create_input_plus_noise(shape, is_complex):
-    x = np.arange(np.product(shape)).reshape(shape)
+    x = np.arange(np.prod(shape)).reshape(shape)
     x = torch.tensor(x, dtype=torch.get_default_dtype())
 
     if is_complex: