account for deprecated numpy.product (use numpy.prod instead)

kaushikcfd · Jul 30, 2023 · 8a0ff10 · 8a0ff10
1 parent fb92a30
commit 8a0ff10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/feinsum/measure.py b/src/feinsum/measure.py
@@ -298,7 +298,7 @@ def _get_footprint_gbytes(expr: FusedEinsum, long_dim_length: int) -> float:
 
     # TODO: mypy is right arg.shape can be 'Any' expression
     return sum(  # type: ignore[no-any-return]
-        np.product(arg.shape) * arg.dtype.itemsize
+        np.prod(arg.shape) * arg.dtype.itemsize
         for arg in kernel.args) * 1e-9