Merge pull request datajoint#167 from sidhulyalkar/modular_clustering

kushalbakshi · Jul 18, 2023 · 8ad337c · 8ad337c
2 parents d7b7738 + deac3cf
commit 8ad337c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/element_array_ephys/spike_sorting/ecephys_spike_sorting.py b/element_array_ephys/spike_sorting/ecephys_spike_sorting.py
@@ -211,7 +211,7 @@ def make(self, key):
                 run_CatGT=run_CatGT,
             )
             # Include so catGT is not rerun when modules are run
-            run_kilosort._CatGT_finished = True
+            run_kilosort._CatGT_finished = run_CatGT
             run_kilosort.run_modules(modules_to_run)
         elif acq_software == "Open Ephys":
             oe_probe = ephys.get_openephys_probe_data(key)
@@ -288,7 +288,7 @@ def make(self, key):
                 KS2ver=f'{Decimal(clustering_method.replace("kilosort", "")):.1f}',
                 run_CatGT=run_CatGT,
             )
-            run_kilosort._CatGT_finished = True
+            run_kilosort._CatGT_finished = run_CatGT
             run_kilosort.run_modules(modules_to_run)
 
         elif acq_software == "Open Ephys":
@@ -337,4 +337,4 @@ def _get_execution_duration(modules_input_hash_fp, modules_to_run):
         dur = modules_status[m]["duration"] or 0
         logged_exec_dur += int(dur)
 
-    return logged_exec_dur, modules_status
+    return logged_exec_dur, modules_status