From 74b1c11bb950c2466de100ceeac499378d16a9cf Mon Sep 17 00:00:00 2001 From: zm711 <92116279+zm711@users.noreply.github.com> Date: Fri, 12 Apr 2024 16:15:53 -0400 Subject: [PATCH] fix broken sorting analyzer example --- .../modules_gallery/core/plot_4_sorting_analyzer.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/examples/modules_gallery/core/plot_4_sorting_analyzer.py b/examples/modules_gallery/core/plot_4_sorting_analyzer.py index b41109bf6a..f38746bbba 100644 --- a/examples/modules_gallery/core/plot_4_sorting_analyzer.py +++ b/examples/modules_gallery/core/plot_4_sorting_analyzer.py @@ -72,7 +72,12 @@ # when using format="binary_folder" or format="zarr" folder = "analyzer_folder" -analyzer = create_sorting_analyzer(sorting=sorting, recording=recording, format="binary_folder", folder=folder) +analyzer = create_sorting_analyzer(sorting=sorting, + recording=recording, + format="binary_folder", + return_scaled=True, # this is the default to attempt to return scaled + folder=folder + ) print(analyzer) # then it can be loaded back @@ -90,7 +95,7 @@ method="uniform", max_spikes_per_unit=500, ) -analyzer.compute("waveforms", ms_before=1.0, ms_after=2.0, return_scaled=True) +analyzer.compute("waveforms", ms_before=1.0, ms_after=2.0) analyzer.compute("templates", operators=["average", "median", "std"]) print(analyzer) @@ -100,12 +105,12 @@ # using parallel processing (recommended!). Like this analyzer.compute( - "waveforms", ms_before=1.0, ms_after=2.0, return_scaled=True, n_jobs=8, chunk_duration="1s", progress_bar=True + "waveforms", ms_before=1.0, ms_after=2.0, n_jobs=8, chunk_duration="1s", progress_bar=True ) # which is equivalent to this: job_kwargs = dict(n_jobs=8, chunk_duration="1s", progress_bar=True) -analyzer.compute("waveforms", ms_before=1.0, ms_after=2.0, return_scaled=True, **job_kwargs) +analyzer.compute("waveforms", ms_before=1.0, ms_after=2.0, **job_kwargs) ################################################################################# # Because certain extensions rely on others (e.g. we need waveforms to calculate