Fix product integer overflow (#193)

czbiohub-sf · Oct 24, 2023 · e822aa8 · e822aa8
1 parent de79ed3
commit e822aa8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/iohub/convert.py b/iohub/convert.py
@@ -357,7 +357,8 @@ def _gen_chunks(self, input_chunks):
         # it's OK if a single image is larger than MAX_CHUNK_SIZE
         while (
             chunks[-3] > 1
-            and np.prod(chunks) * bytes_per_pixel > MAX_CHUNK_SIZE
+            and np.prod(chunks, dtype=np.int64) * bytes_per_pixel
+            > MAX_CHUNK_SIZE
         ):
             chunks[-3] = np.ceil(chunks[-3] / 2).astype(int)