Skip to content

Commit 87e6b0c

Browse files
committed
Solve conflicts 2
2 parents 1b8aa4b + 10e2f23 commit 87e6b0c

4 files changed

Lines changed: 7 additions & 4 deletions

File tree

src/spikeinterface/sorters/internal/spyking_circus2.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class Spykingcircus2Sorter(ComponentsBasedSorter):
3434
"apply_motion_correction": True,
3535
"motion_correction": {"preset": "dredge_fast"},
3636
"merging": {"max_distance_um": 50},
37-
"clustering": {"method": "circus-clustering", "method_kwargs": dict(remove_small_snr=True)},
37+
"clustering": {"method": "circus", "method_kwargs": dict()},
3838
"matching": {"method": "circus-omp-svd", "method_kwargs": dict(), "gather_mode": "memory"},
3939
"apply_preprocessing": True,
4040
"templates_from_svd": True,
@@ -386,8 +386,9 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
386386
matching_params["templates"] = templates
387387

388388
if matching_method is not None:
389+
gather_kwargs = {}
389390
if gather_mode == "npy":
390-
gather_kwargs["folder"] = gather_kwargs.get("folder", sorter_output_folder / "matching")
391+
gather_kwargs["folder"] = sorter_output_folder / "matching"
391392
spikes = find_spikes_from_templates(
392393
recording_w,
393394
matching_method,

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
class SpykingCircus2SorterCommonTestSuite(SorterCommonTestSuite, unittest.TestCase):
1010
SorterClass = Spykingcircus2Sorter
1111

12+
@unittest.skip("performance reason")
1213
def test_with_numpy_gather(self):
1314
recording = self.recording
1415
sorter_name = self.SorterClass.sorter_name

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
class Tridesclous2SorterCommonTestSuite(SorterCommonTestSuite, unittest.TestCase):
1111
SorterClass = Tridesclous2Sorter
1212

13+
@unittest.skip("performance reason")
1314
def test_with_numpy_gather(self):
1415
recording = self.recording
1516
sorter_name = self.SorterClass.sorter_name

src/spikeinterface/sorters/internal/tridesclous2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -278,13 +278,13 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
278278
## peeler
279279
matching_method = params["matching"].pop("method")
280280
gather_mode = params["matching"].pop("gather_mode", "memory")
281-
gather_kwargs = params["matching"].pop("gather_kwargs", {})
282281
matching_params = params["matching"].get("matching_kwargs", {}).copy()
283282
matching_params["templates"] = templates
284283
if matching_method in ("tdc-peeler",):
285284
matching_params["noise_levels"] = noise_levels
285+
gather_kwargs = {}
286286
if gather_mode == "npy":
287-
gather_kwargs = {"folder": gather_kwargs.get("folder", sorter_output_folder / "matching")}
287+
gather_kwargs["folder"] = sorter_output_folder / "matching"
288288
spikes = find_spikes_from_templates(
289289
recording_for_peeler,
290290
method=matching_method,

0 commit comments

Comments
 (0)