Upcasting raw mat dtype resolves #883

int-brain-lab · Nov 23, 2024 · 136d6e0 · 136d6e0
1 parent 00f9f4c
commit 136d6e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ibllib/__init__.py b/ibllib/__init__.py
@@ -2,7 +2,7 @@
 import logging
 import warnings
 
-__version__ = '3.0rc0'
+__version__ = '3.0rc1'
 warnings.filterwarnings('always', category=DeprecationWarning, module='ibllib')
 
 # if this becomes a full-blown library we should let the logging configuration to the discretion of the dev

diff --git a/ibllib/pipes/mesoscope_tasks.py b/ibllib/pipes/mesoscope_tasks.py
@@ -796,8 +796,8 @@ def _run(self, rename_files=True, use_badframes=True, **kwargs):
         for m, collection in zip(all_meta, raw_image_collections):
             badframes_path = self.session_path.joinpath(collection, 'badframes.mat')
             if badframes_path.exists():
-                raw_mat = loadmat(badframes_path, squeeze_me=True, simplify_cells=True)['badframes']
-                badframes = np.r_[badframes, raw_mat + total_frames]
+                raw_mat = loadmat(badframes_path, squeeze_me=True, simplify_cells=True)
+                badframes = np.r_[badframes, raw_mat['badframes'].astype('uint32') + total_frames]
             total_frames += m['nFrames']
         if len(badframes) > 0 and use_badframes is True:
             # The badframes array should always be a subset of the frameQC array