Fix invalid indexing for python <3.11

flatironinstitute · Mar 29, 2024 · 6aedb79 · 6aedb79
1 parent ae20ae2
commit 6aedb79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caiman/utils/sbx_utils.py b/caiman/utils/sbx_utils.py
@@ -432,7 +432,7 @@ def get_chunk(n: int):
             chunk = np.transpose(chunk, (0, 4, 2, 1, 3))[channel]  # to (frames, Y, X, Z)
             if not is3D:
                 chunk = np.squeeze(chunk, axis=3)
-            return chunk[:, *np.ix_(*subindices[1:])]
+            return chunk[(slice(None),) + np.ix_(*subindices[1:])]
 
         if frame_stride == 1:
             sbx_file.seek(subindices[0][0] * frame_size, 0)