From db8997141d0ec9ea165413ee9e2d1a2a1cd0ebbf Mon Sep 17 00:00:00 2001 From: Ji Hoon Jeong Date: Mon, 1 Apr 2024 13:22:35 -0400 Subject: [PATCH] force integer value for `np.arange` when indexing h5 file with 4D and 5D data be8830 --- suite2p/io/h5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/suite2p/io/h5.py b/suite2p/io/h5.py index fbfe53e8..079c9654 100644 --- 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