Force dtype of prod in _ravel_shape_indices (#57)

Force dtype of `prod` in `_ravel_shape_indices`
dask · Sep 3, 2018 · 8db18e9 · 8db18e9
1 parent aab8434
commit 8db18e9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dask_image/ndmeasure/_utils.py b/dask_image/ndmeasure/_utils.py
@@ -57,8 +57,11 @@ def _ravel_shape_indices(dimensions, dtype=int, chunks=None):
 
     indices = sum([
         dask.array.arange(
-            0, numpy.prod(dimensions[i:]), numpy.prod(dimensions[i + 1:]),
-            dtype=dtype, chunks=c
+            0,
+            numpy.prod(dimensions[i:], dtype=dtype),
+            numpy.prod(dimensions[i + 1:], dtype=dtype),
+            dtype=dtype,
+            chunks=c
         )[i * (None,) + (slice(None),) + (len(dimensions) - i - 1) * (None,)]
         for i, c in enumerate(chunks)
     ])