bugfix in updating median-subtraction duration

Author: ttngu207

element_array_ephys/readers/kilosort_triggering.py

@@ -347,6 +347,7 @@ def __init__(self, npx_input_dir: str, ks_output_dir: str,
         self._json_directory = self._ks_output_dir / 'json_configs'
         self._json_directory.mkdir(parents=True, exist_ok=True)
 
+        self._median_subtraction_status = {}
         self.ks_input_params = None
         self._modules_input_hash = None
         self._modules_input_hash_fp = None
@@ -427,6 +428,13 @@ def run_modules(self):
             if module_status['completion_time'] is not None:
                 continue
 
+            if module == 'median_subtraction' and self._median_subtraction_status:
+                median_subtraction_status = self._get_module_status('median_subtraction')
+                median_subtraction_status['duration'] = self._median_subtraction_status['duration']
+                median_subtraction_status['completion_time'] = datetime.strptime(median_subtraction_status['start_time'], '%Y-%m-%d %H:%M:%S.%f') + timedelta(seconds=median_subtraction_status['duration'])
+                self._update_module_status({'median_subtraction': median_subtraction_status})
+                continue
+
             module_output_json = self._get_module_output_json_filename(module)
             command = [sys.executable,
                     '-W', 'ignore', '-m', 'ecephys_spike_sorting.modules.' + module,
@@ -470,10 +478,7 @@ def _get_raw_data_filepaths(self):
                         if (line.startswith('ecephys spike sorting: median subtraction module')
                                 and previous_line.startswith('Total processing time:')):
                             duration = int(re.search('\d+\.?\d+', previous_line).group())
-                            median_subtraction_status = self._get_module_status('median_subtraction')
-                            median_subtraction_status['duration'] = duration
-                            median_subtraction_status['completion_time'] = datetime.strptime(median_subtraction_status['start_time'], '%Y-%m-%d %H:%M:%S.%f') + timedelta(seconds=duration)
-                            self._update_module_status({'median_subtraction': median_subtraction_status})
+                            self._median_subtraction_status['duration'] = duration
                             return continuous_file
                         previous_line = line