fix waveform extactor with empty sorting and sparse

SpikeInterface · Oct 4, 2023 · 204c8e9 · 204c8e9
1 parent 590cd6b
commit 204c8e9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/spikeinterface/core/sparsity.py b/src/spikeinterface/core/sparsity.py
@@ -102,7 +102,11 @@ def __init__(self, mask, unit_ids, channel_ids):
 
         self.num_channels = self.channel_ids.size
         self.num_units = self.unit_ids.size
-        self.max_num_active_channels = self.mask.sum(axis=1).max()
+        if self.mask.shape[0]:
+            self.max_num_active_channels = self.mask.sum(axis=1).max()
+        else:
+            # empty sorting without units
+            self.max_num_active_channels = 0
 
     def __repr__(self):
         density = np.mean(self.mask)

diff --git a/src/spikeinterface/core/tests/test_waveform_extractor.py b/src/spikeinterface/core/tests/test_waveform_extractor.py
@@ -556,4 +556,5 @@ def test_non_json_object():
     # test_portability()
     # test_recordingless()
     # test_compute_sparsity()
-    test_non_json_object()
+    # test_non_json_object()
+    test_empty_sorting()