Skip to content

Commit 86a152c

Browse files
authored
Merge pull request #3262 from mhhennig/main
added lowpass parameter, fixed verbose option
2 parents f5e6bea + 0b7dc33 commit 86a152c

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/spikeinterface/sorters/external/herdingspikes.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ class HerdingspikesSorter(BaseSorter):
1919
"chunk_size": None,
2020
"rescale": True,
2121
"rescale_value": -1280.0,
22+
"lowpass": True,
2223
"common_reference": "median",
2324
"spike_duration": 1.0,
2425
"amp_avg_duration": 0.4,
@@ -53,6 +54,7 @@ class HerdingspikesSorter(BaseSorter):
5354
"out_file": "Path and filename to store detection and clustering results. (`str`, `HS2_detected`)",
5455
"verbose": "Print progress information. (`bool`, `True`)",
5556
"chunk_size": " Number of samples per chunk during detection. If `None`, a suitable value will be estimated. (`int`, `None`)",
57+
"lowpass": "Enable internal low-pass filtering (simple two-step average). (`bool`, `True`)",
5658
"common_reference": "Method for common reference filtering, can be `average` or `median` (`str`, `median`)",
5759
"rescale": "Automatically re-scale the data. (`bool`, `True`)",
5860
"rescale_value": "Factor by which data is re-scaled. (`float`, `-1280.0`)",
@@ -122,20 +124,21 @@ def _run_from_folder(cls, sorter_output_folder, params, verbose):
122124

123125
hs_version = version.parse(hs.__version__)
124126

125-
if hs_version >= version.parse("0.4.001"):
127+
if hs_version >= version.parse("0.4.1"):
126128
lightning_api = True
127129
else:
128130
lightning_api = False
129131

130132
assert (
131133
lightning_api
132-
), "HerdingSpikes version <0.4.001 is no longer supported. run:\n>>> pip install --upgrade herdingspikes"
134+
), "HerdingSpikes version <0.4.1 is no longer supported. To upgrade, run:\n>>> pip install --upgrade herdingspikes"
133135

134136
recording = cls.load_recording_from_folder(sorter_output_folder.parent, with_warnings=False)
135137

136138
sorted_file = str(sorter_output_folder / "HS2_sorted.hdf5")
137139
params["out_file"] = str(sorter_output_folder / "HS2_detected")
138140
p = params
141+
p.update({"verbose": verbose})
139142

140143
det = hs.HSDetectionLightning(recording, p)
141144
det.DetectFromRaw()

0 commit comments

Comments
 (0)