Skip to content

Commit 7e23776

Browse files
authored
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
1 parent 7039bc1 commit 7e23776

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/conftest.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def test_params():
2626

2727

2828
def create_input_plus_noise(shape, is_complex):
29-
x = np.arange(np.product(shape)).reshape(shape)
29+
x = np.arange(np.prod(shape)).reshape(shape)
3030
x = torch.tensor(x, dtype=torch.get_default_dtype())
3131

3232
if is_complex:

0 commit comments

Comments
 (0)