Initialize dataset

cwindolf · Oct 15, 2024 · 3421c8f · 3421c8f
1 parent 19da675
commit 3421c8f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/dartsort/peel/peel_base.py b/src/dartsort/peel/peel_base.py
@@ -685,9 +685,7 @@ def initialize_files(
                     )
 
         if residual_to_h5:
-            if "residual" in output_h5:
-                pass
-            else:
+            if "residual" not in output_h5:
                 n_chans = self.recording.get_num_channels()
                 output_h5.create_dataset(
                     "residual",
@@ -696,6 +694,14 @@ def initialize_files(
                     maxshape=(None, self.spike_length_samples, n_chans),
                     chunks=(chunk_size, self.spike_length_samples, n_chans),
                 )
+            if "residual_times_seconds" not in output_h5:
+                output_h5.create_dataset(
+                    "residual_times_seconds",
+                    dtype=float,
+                    shape=(0,),
+                    maxshape=(None,),
+                    chunks=(chunk_size,),
+                )
 
         # residual file ignore/open/overwrite logic
         save_residual = residual_filename is not None