Skip to content

Commit 066caa0

Browse files
samuelgarciasamuelgarcia
committed
Move components benchmarks
1 parent 6cff8fa commit 066caa0

17 files changed

Lines changed: 8 additions & 25 deletions

src/spikeinterface/sortingcomponents/benchmark/benchmark_clustering.py renamed to src/spikeinterface/benchmark/benchmark_clustering.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,7 @@
1111

1212
import numpy as np
1313

14-
15-
from .benchmark_tools import BenchmarkStudy, Benchmark
14+
from .benchmark_base import Benchmark, BenchmarkStudy
1615
from spikeinterface.core.sortinganalyzer import create_sorting_analyzer
1716
from spikeinterface.core.template_tools import get_template_extremum_channel
1817

src/spikeinterface/sortingcomponents/benchmark/benchmark_matching.py renamed to src/spikeinterface/benchmark/benchmark_matching.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,8 @@
99
)
1010

1111
import numpy as np
12-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy
12+
from .benchmark_base import Benchmark, BenchmarkStudy
1313
from spikeinterface.core.basesorting import minimum_spike_dtype
14-
from spikeinterface.sortingcomponents.tools import remove_empty_templates
15-
from spikeinterface.core.recording_tools import get_noise_levels
16-
from spikeinterface.core.sparsity import compute_sparsity
1714

1815

1916
class MatchingBenchmark(Benchmark):

src/spikeinterface/sortingcomponents/benchmark/benchmark_motion_estimation.py renamed to src/spikeinterface/benchmark/benchmark_motion_estimation.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,8 @@
88
import numpy as np
99

1010
from spikeinterface.core import get_noise_levels
11-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy, _simpleaxis
11+
from .benchmark_base import Benchmark, BenchmarkStudy
12+
from .benchmark_plot_tools import _simpleaxis
1213
from spikeinterface.sortingcomponents.motion import estimate_motion
1314
from spikeinterface.sortingcomponents.peak_detection import detect_peaks
1415
from spikeinterface.sortingcomponents.peak_selection import select_peaks

src/spikeinterface/sortingcomponents/benchmark/benchmark_motion_interpolation.py renamed to src/spikeinterface/benchmark/benchmark_motion_interpolation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from spikeinterface.curation import MergeUnitsSorting
1111

1212

13-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy, _simpleaxis
13+
from .benchmark_base import Benchmark, BenchmarkStudy
1414

1515

1616
class MotionInterpolationBenchmark(Benchmark):

src/spikeinterface/sortingcomponents/benchmark/benchmark_peak_detection.py renamed to src/spikeinterface/benchmark/benchmark_peak_detection.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,9 @@
1212

1313
import numpy as np
1414

15-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy
15+
from .benchmark_base import Benchmark, BenchmarkStudy
1616
from spikeinterface.core.basesorting import minimum_spike_dtype
1717
from spikeinterface.core.sortinganalyzer import create_sorting_analyzer
18-
from spikeinterface.core.template_tools import get_template_extremum_channel
1918

2019

2120
class PeakDetectionBenchmark(Benchmark):

src/spikeinterface/sortingcomponents/benchmark/benchmark_peak_localization.py renamed to src/spikeinterface/benchmark/benchmark_peak_localization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
compute_grid_convolution,
77
)
88
import numpy as np
9-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy
9+
from .benchmark_base import Benchmark, BenchmarkStudy
1010
from spikeinterface.core.sortinganalyzer import create_sorting_analyzer
1111

1212

src/spikeinterface/sortingcomponents/benchmark/benchmark_peak_selection.py renamed to src/spikeinterface/benchmark/benchmark_peak_selection.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,16 +6,9 @@
66
from spikeinterface.comparison.comparisontools import make_matching_events
77
from spikeinterface.core import get_noise_levels
88

9-
import time
10-
import string, random
11-
import pylab as plt
12-
import os
139
import numpy as np
1410

15-
from spikeinterface.sortingcomponents.benchmark.benchmark_tools import Benchmark, BenchmarkStudy
16-
from spikeinterface.core.basesorting import minimum_spike_dtype
17-
from spikeinterface.core.sortinganalyzer import create_sorting_analyzer
18-
11+
from .benchmark_base import Benchmark, BenchmarkStudy
1912

2013
class PeakSelectionBenchmark(Benchmark):
2114

src/spikeinterface/sortingcomponents/benchmark/benchmark_tools.py renamed to src/spikeinterface/benchmark/benchmark_tools.py

File renamed without changes.

src/spikeinterface/sortingcomponents/benchmark/tests/common_benchmark_testing.py renamed to src/spikeinterface/benchmark/tests/common_benchmark_testing.py

File renamed without changes.

src/spikeinterface/sortingcomponents/benchmark/tests/test_benchmark_clustering.py renamed to src/spikeinterface/benchmark/tests/test_benchmark_clustering.py

File renamed without changes.

0 commit comments

Comments
 (0)