Fix gpuci: np.product is deprecated (#8518)

dask · Feb 22, 2024 · e6fe6f2 · e6fe6f2
1 parent a5a6e99
commit e6fe6f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/distributed/protocol/tests/test_cupy.py b/distributed/protocol/tests/test_cupy.py
@@ -17,7 +17,7 @@
 @pytest.mark.parametrize("order", ["C", "F"])
 @pytest.mark.parametrize("serializers", [("cuda",), ("dask",), ("pickle",)])
 def test_serialize_cupy(shape, dtype, order, serializers):
-    x = cupy.arange(numpy.product(shape), dtype=dtype)
+    x = cupy.arange(numpy.prod(shape), dtype=dtype)
     x = cupy.ndarray(shape, dtype=x.dtype, memptr=x.data, order=order)
     header, frames = serialize(x, serializers=serializers)
     y = deserialize(header, frames, deserializers=serializers)

diff --git a/distributed/protocol/tests/test_numba.py b/distributed/protocol/tests/test_numba.py
@@ -20,7 +20,7 @@ def test_serialize_numba(shape, dtype, order, serializers):
     if not cuda.is_available():
         pytest.skip("CUDA is not available")
 
-    ary = np.arange(np.product(shape), dtype=dtype)
+    ary = np.arange(np.prod(shape), dtype=dtype)
     ary = np.ndarray(shape, dtype=ary.dtype, buffer=ary.data, order=order)
     x = cuda.to_device(ary)
     header, frames = serialize(x, serializers=serializers)