Skip to content

Commit d1d65f6

Browse files
committed
estimate_sparsity arg ordering
1 parent 790715c commit d1d65f6

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

src/spikeinterface/core/sortinganalyzer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def create_sorting_analyzer(
127127
recording.channel_ids, sparsity.channel_ids
128128
), "create_sorting_analyzer(): if external sparsity is given unit_ids must correspond"
129129
elif sparse:
130-
sparsity = estimate_sparsity(recording, sorting, **sparsity_kwargs)
130+
sparsity = estimate_sparsity(sorting, recording, **sparsity_kwargs)
131131
else:
132132
sparsity = None
133133

src/spikeinterface/core/sparsity.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -539,8 +539,8 @@ def compute_sparsity(
539539

540540

541541
def estimate_sparsity(
542-
recording: BaseRecording,
543542
sorting: BaseSorting,
543+
recording: BaseRecording,
544544
num_spikes_for_sparsity: int = 100,
545545
ms_before: float = 1.0,
546546
ms_after: float = 2.5,
@@ -563,10 +563,10 @@ def estimate_sparsity(
563563
564564
Parameters
565565
----------
566-
recording: BaseRecording
567-
The recording
568566
sorting: BaseSorting
569567
The sorting
568+
recording: BaseRecording
569+
The recording
570570
num_spikes_for_sparsity: int, default: 100
571571
How many spikes per units to compute the sparsity
572572
ms_before: float, default: 1.0

src/spikeinterface/core/tests/test_sparsity.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,8 +166,8 @@ def test_estimate_sparsity():
166166

167167
# small radius should give a very sparse = one channel per unit
168168
sparsity = estimate_sparsity(
169-
recording,
170169
sorting,
170+
recording,
171171
num_spikes_for_sparsity=50,
172172
ms_before=1.0,
173173
ms_after=2.0,
@@ -182,8 +182,8 @@ def test_estimate_sparsity():
182182

183183
# best_channel : the mask should exactly 3 channels per units
184184
sparsity = estimate_sparsity(
185-
recording,
186185
sorting,
186+
recording,
187187
num_spikes_for_sparsity=50,
188188
ms_before=1.0,
189189
ms_after=2.0,

src/spikeinterface/postprocessing/tests/common_extension_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ class AnalyzerExtensionCommonTestSuite:
7979
def setUpClass(cls):
8080
cls.recording, cls.sorting = get_dataset()
8181
# sparsity is computed once for all cases to save processing time and force a small radius
82-
cls.sparsity = estimate_sparsity(cls.recording, cls.sorting, method="radius", radius_um=20)
82+
cls.sparsity = estimate_sparsity(cls.sorting, cls.recording, method="radius", radius_um=20)
8383

8484
@property
8585
def extension_name(self):

0 commit comments

Comments
 (0)