From 4c9afe8125eb96251b10a1200b8f21c4165ce4c5 Mon Sep 17 00:00:00 2001 From: Nicholas Watters Date: Tue, 12 Aug 2025 16:17:32 -0400 Subject: [PATCH] Update medicine.py to expose additional arguments The existing implementation does not expose the "output_dir" and "plot_figures" arguments, but these arguments are important for debugging. This change exposes those arguments. --- src/spikeinterface/sortingcomponents/motion/medicine.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/spikeinterface/sortingcomponents/motion/medicine.py b/src/spikeinterface/sortingcomponents/motion/medicine.py index b58b836718..8349a79369 100644 --- a/src/spikeinterface/sortingcomponents/motion/medicine.py +++ b/src/spikeinterface/sortingcomponents/motion/medicine.py @@ -35,6 +35,8 @@ def run( extra, bin_s=1.0, ## medicine specific kwargs propagated to the lib + output_dir=None, + plot_figures=False, motion_bound=800, time_kernel_width=30, activity_network_hidden_features=(256, 256), @@ -78,8 +80,8 @@ def run( peak_times=peaks["sample_index"] / recording.get_sampling_frequency(), time_bin_size=bin_s, num_depth_bins=num_depth_bins, - output_dir=None, - plot_figures=False, + output_dir=output_dir, + plot_figures=plot_figures, motion_bound=motion_bound, time_kernel_width=time_kernel_width, activity_network_hidden_features=activity_network_hidden_features,