force integer value for np.arange when indexing h5 file with 4D and…

… 5D data be8830
MouseLand · Apr 1, 2024 · db89971 · db89971
1 parent be8830d
commit db89971
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/suite2p/io/h5.py b/suite2p/io/h5.py
@@ -74,7 +74,7 @@ def h5py_to_binary(ops):
                         im = f[key][irange, :, :]
                     else:
                         irange = np.arange(
-                            ik / ncp, min(ik / ncp + nbatch / ncp, nframes_all / ncp),
+                            int(ik / ncp), int(min(ik / ncp + nbatch / ncp, nframes_all / ncp)),
                             1)
                         if irange.size == 0:
                             break