Skip to content

Commit 78f732f

Browse files
committed
Rename threshold_metrics_label_units
1 parent 57616d0 commit 78f732f

4 files changed

Lines changed: 10 additions & 9 deletions

File tree

doc/api.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -373,6 +373,7 @@ spikeinterface.curation
373373
.. autofunction:: remove_redundant_units
374374
.. autofunction:: remove_duplicated_spikes
375375
.. autofunction:: remove_excess_spikes
376+
.. autofunction:: threshold_metrics_label_units
376377
.. autofunction:: model_based_label_units
377378
.. autofunction:: load_model
378379
.. autofunction:: train_model

src/spikeinterface/curation/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from .sortingview_curation import apply_sortingview_curation
2121

2222
# automated curation
23-
from .qualitymetrics_curation import qualitymetrics_label_units
23+
from .qualitymetrics_curation import threshold_metrics_label_units
2424
from .model_based_curation import model_based_label_units, load_model, auto_label_units
2525
from .train_manual_curation import train_model, get_default_classifier_search_spaces
2626
from .unitrefine_curation import unitrefine_label_units

src/spikeinterface/curation/qualitymetrics_curation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from .curation_tools import is_threshold_disabled
99

1010

11-
def qualitymetrics_label_units(
11+
def threshold_metrics_label_units(
1212
analyzer: SortingAnalyzer,
1313
thresholds: dict | str | Path,
1414
):

src/spikeinterface/curation/tests/test_qualitymetrics_curation.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,19 +2,19 @@
22
import json
33

44
from spikeinterface.curation.tests.common import sorting_analyzer_for_curation
5-
from spikeinterface.curation import qualitymetrics_label_units
5+
from spikeinterface.curation import threshold_metrics_label_units
66

77

8-
def test_qualitymetrics_label_units(sorting_analyzer_for_curation):
9-
"""Test the `qualitymetrics_label_units` function."""
8+
def test_threshold_metrics_label_units(sorting_analyzer_for_curation):
9+
"""Test the `threshold_metrics_label_units` function."""
1010
sorting_analyzer_for_curation.compute("quality_metrics")
1111

1212
thresholds = {
1313
"snr": {"min": 5.0},
1414
"firing_rate": {"min": 0.1, "max": 20.0},
1515
}
1616

17-
labels = qualitymetrics_label_units(
17+
labels = threshold_metrics_label_units(
1818
sorting_analyzer_for_curation,
1919
thresholds,
2020
)
@@ -36,8 +36,8 @@ def test_qualitymetrics_label_units(sorting_analyzer_for_curation):
3636
assert labels.loc[unit_id, "label"] == "noise"
3737

3838

39-
def test_qualitymetrics_label_units_with_file(sorting_analyzer_for_curation, tmp_path):
40-
"""Test the `qualitymetrics_label_units` function with thresholds from a JSON file."""
39+
def test_threshold_metrics_label_units_with_file(sorting_analyzer_for_curation, tmp_path):
40+
"""Test the `threshold_metrics_label_units` function with thresholds from a JSON file."""
4141
sorting_analyzer_for_curation.compute("quality_metrics")
4242

4343
thresholds = {
@@ -49,7 +49,7 @@ def test_qualitymetrics_label_units_with_file(sorting_analyzer_for_curation, tmp
4949
with open(thresholds_file, "w") as f:
5050
json.dump(thresholds, f)
5151

52-
labels = qualitymetrics_label_units(
52+
labels = threshold_metrics_label_units(
5353
sorting_analyzer_for_curation,
5454
thresholds_file,
5555
)

0 commit comments

Comments
 (0)