clean and debug

Author: samuelgarcia

src/spikeinterface/sorters/internal/tests/test_spykingcircus2.py

@@ -4,12 +4,16 @@
 
 from spikeinterface.sorters import Spykingcircus2Sorter
 
+from pathlib import Path
 
 class SpykingCircus2SorterCommonTestSuite(SorterCommonTestSuite, unittest.TestCase):
     SorterClass = Spykingcircus2Sorter
 
 
 if __name__ == "__main__":
+    from spikeinterface import set_global_job_kwargs
+    set_global_job_kwargs(n_jobs=1, progress_bar=False)
     test = SpykingCircus2SorterCommonTestSuite()
+    test.cache_folder = Path(__file__).resolve().parents[4] / "cache_folder" / "sorters"
     test.setUp()
     test.test_with_run()

src/spikeinterface/sorters/internal/tests/test_tridesclous2.py

@@ -4,12 +4,15 @@
 
 from spikeinterface.sorters import Tridesclous2Sorter
 
+from pathlib import Path
+
 
 class Tridesclous2SorterCommonTestSuite(SorterCommonTestSuite, unittest.TestCase):
     SorterClass = Tridesclous2Sorter
 
 
 if __name__ == "__main__":
     test = Tridesclous2SorterCommonTestSuite()
+    test.cache_folder = Path(__file__).resolve().parents[4] / "cache_folder" / "sorters"
     test.setUp()
     test.test_with_run()

src/spikeinterface/sorters/internal/tridesclous2.py

@@ -226,7 +226,8 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
         matching_method = params["matching"]["method"]
         matching_params = params["matching"]["method_kwargs"].copy()
         matching_params["templates"] = templates
-        matching_params["noise_levels"] = noise_levels
+        if params["matching"]["method"] in ("tdc-peeler", ):
+            matching_params["noise_levels"] = noise_levels
         spikes = find_spikes_from_templates(
             recording_for_peeler, method=matching_method, method_kwargs=matching_params, **job_kwargs
         )

src/spikeinterface/sortingcomponents/matching/base.py

@@ -32,11 +32,12 @@ def compute(self, traces, start_frame, end_frame, segment_index, max_margin):
         spikes["segment_index"] = segment_index
 
         margin = self.get_trace_margin()
-        if margin > 0:
+        if margin > 0 and spikes.size > 0:
             keep = (spikes["sample_index"] >= margin) & (spikes["sample_index"] < (traces.shape[0] - margin))
             spikes = spikes[keep]
 
-        return spikes
+        # node pipeline need to return a tuple
+        return (spikes, )
 
     def compute_matching(self, traces, start_frame, end_frame, segment_index):
         raise NotImplementedError