Fixed bug introduced with logging for nblocks=0

MouseLand · Jun 8, 2024 · 2e9d0f0 · 2e9d0f0
1 parent fc78f34
commit 2e9d0f0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/kilosort/run_kilosort.py b/kilosort/run_kilosort.py
@@ -402,10 +402,11 @@ def compute_drift_correction(ops, device, tic0=np.nan, progress_bar=None,
     bfile.close()
     logger.info(f'drift computed in {time.time()-tic : .2f}s; ' + 
                 f'total {time.time()-tic0 : .2f}s')
-    logger.debug(f'st shape: {st.shape}')
-    logger.debug(f'yblk shape: {ops["yblk"].shape}')
-    logger.debug(f'dshift shape: {ops["dshift"].shape}')
-    logger.debug(f'iKxx shape: {ops["iKxx"].shape}')
+    if st is not None:
+        logger.debug(f'st shape: {st.shape}')
+        logger.debug(f'yblk shape: {ops["yblk"].shape}')
+        logger.debug(f'dshift shape: {ops["dshift"].shape}')
+        logger.debug(f'iKxx shape: {ops["iKxx"].shape}')
 
     # binary file with drift correction
     bfile = io.BinaryFiltered(