Fix int overflow bug due to older numpy version

neuroinformatics-unit · Jul 25, 2024 · 58678cc · 58678cc
1 parent cb98655
commit 58678cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/derotation/load_data/custom_data_loaders.py b/derotation/load_data/custom_data_loaders.py
@@ -67,7 +67,8 @@ def get_analog_signals(path_to_aux: str, channel_names: list) -> tuple:
         _description_
     """
 
-    data = np.fromfile(path_to_aux, dtype=np.int16)
+    data = np.fromfile(path_to_aux, dtype=np.int16)  # Has to be read as int16
+    data = data.astype(np.int32)  # cast data to int32 to avoid overflow
     data = data.reshape((-1, len(channel_names)))
     data = convert_to_volts(data)