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,