oups

SpikeInterface · Oct 25, 2023 · 3ed9e5f · 3ed9e5f
1 parent 59f0473
commit 3ed9e5f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/spikeinterface/postprocessing/spike_locations.py b/src/spikeinterface/postprocessing/spike_locations.py
@@ -33,7 +33,7 @@ def _set_params(
         spike_retriver_kwargs=dict(
             channel_from_template=True,
             radius_um=50,
-            peaks_sign="neg",
+            peak_sign="neg",
         ),
         method="center_of_mass",
         method_kwargs={},
@@ -127,7 +127,7 @@ def compute_spike_locations(
     spike_retriver_kwargs=dict(
         channel_from_template=True,
         radius_um=50,
-        peaks_sign="neg",
+        peak_sign="neg",
     ),
     method="center_of_mass",
     method_kwargs={},
@@ -156,7 +156,7 @@ def compute_spike_locations(
                 channel_from_template = False is slower but more accurate
             * radius_um: float, default 50
                 In case channel_from_template=False, this is the radius to get the true peak.
-            * peaks_sign="neg"
+            * peak_sign="neg"
                 In case channel_from_template=False, this is the peak sign.
     method : str
         'center_of_mass' / 'monopolar_triangulation' / 'grid_convolution'

diff --git a/src/spikeinterface/postprocessing/tests/test_spike_locations.py b/src/spikeinterface/postprocessing/tests/test_spike_locations.py
@@ -10,8 +10,8 @@ class SpikeLocationsExtensionTest(WaveformExtensionCommonTestSuite, unittest.Tes
     extension_class = SpikeLocationsCalculator
     extension_data_names = ["spike_locations"]
     extension_function_kwargs_list = [
-        dict(method="center_of_mass", chunk_size=10000, n_jobs=1, channel_from_template=True),
-        dict(method="center_of_mass", chunk_size=10000, n_jobs=1, channel_from_template=False),
+        dict(method="center_of_mass", chunk_size=10000, n_jobs=1, spike_retriver_kwargs=dict(channel_from_template=True)),
+        dict(method="center_of_mass", chunk_size=10000, n_jobs=1, spike_retriver_kwargs=dict(channel_from_template=False)),
         dict(method="center_of_mass", chunk_size=10000, n_jobs=1, outputs="by_unit"),
         dict(method="monopolar_triangulation", chunk_size=10000, n_jobs=1, outputs="by_unit"),
         dict(method="monopolar_triangulation", chunk_size=10000, n_jobs=1, outputs="by_unit"),