From 4b83e8b52a906826570df49c7bd29a5c26050235 Mon Sep 17 00:00:00 2001 From: Zach McKenzie <92116279+zm711@users.noreply.github.com> Date: Mon, 29 Jul 2024 13:41:09 -0400 Subject: [PATCH] fix docstring and error --- src/spikeinterface/postprocessing/spike_amplitudes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/spikeinterface/postprocessing/spike_amplitudes.py b/src/spikeinterface/postprocessing/spike_amplitudes.py index e82a9e61e4..f158952132 100644 --- a/src/spikeinterface/postprocessing/spike_amplitudes.py +++ b/src/spikeinterface/postprocessing/spike_amplitudes.py @@ -44,8 +44,8 @@ class ComputeSpikeAmplitudes(AnalyzerExtension): The localization method to use method_kwargs : dict, default: dict() Other kwargs depending on the method. - outputs : "concatenated" | "by_unit", default: "concatenated" - The output format + outputs : "numpy" | "by_unit", default: "numpy" + The output format, either concatenated as numpy array or separated on a per unit basis Returns ------- @@ -148,7 +148,7 @@ def _get_data(self, outputs="numpy"): amplitudes_by_units[segment_index][unit_id] = all_amplitudes[inds] return amplitudes_by_units else: - raise ValueError(f"Wrong .get_data(outputs={outputs})") + raise ValueError(f"Wrong .get_data(outputs={outputs}); possibilities are `numpy` or `by_unit`") register_result_extension(ComputeSpikeAmplitudes)