feat(spike_sorting): handle cases when no units/spikes are found

datajoint · Oct 8, 2024 · 463ec52 · 463ec52
1 parent 3ab94dd
commit 463ec52
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/element_array_ephys/spike_sorting/si_spike_sorting.py b/element_array_ephys/spike_sorting/si_spike_sorting.py
@@ -275,11 +275,17 @@ def make(self, key):
 
         analyzer_output_dir = output_dir / sorter_name / "sorting_analyzer"
 
+        has_units = si_sorting.unit_ids.size > 0
+
         @memoized_result(
             uniqueness_dict=postprocessing_params,
             output_directory=analyzer_output_dir,
         )
         def _sorting_analyzer_compute():
+            if not has_units:
+                log.info("No units found in sorting object. Skipping sorting analyzer.")
+                return
+
             # Sorting Analyzer
             sorting_analyzer = si.create_sorting_analyzer(
                 sorting=si_sorting,
@@ -303,6 +309,8 @@ def _sorting_analyzer_compute():
 
         _sorting_analyzer_compute()
 
+        do_si_export = postprocessing_params.get("export_to_phy", False) or postprocessing_params.get("export_report", False)
+
         self.insert1(
             {
                 **key,
@@ -311,8 +319,7 @@ def _sorting_analyzer_compute():
                     datetime.utcnow() - execution_time
                 ).total_seconds()
                 / 3600,
-                "do_si_export": postprocessing_params.get("export_to_phy", False)
-                or postprocessing_params.get("export_report", False),
+                "do_si_export": do_si_export and has_units,
             }
         )