Skip to content

Commit 4c9afe8

Browse files
authored
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.
1 parent 40ad3fb commit 4c9afe8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/spikeinterface/sortingcomponents/motion/medicine.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ def run(
3535
extra,
3636
bin_s=1.0,
3737
## medicine specific kwargs propagated to the lib
38+
output_dir=None,
39+
plot_figures=False,
3840
motion_bound=800,
3941
time_kernel_width=30,
4042
activity_network_hidden_features=(256, 256),
@@ -78,8 +80,8 @@ def run(
7880
peak_times=peaks["sample_index"] / recording.get_sampling_frequency(),
7981
time_bin_size=bin_s,
8082
num_depth_bins=num_depth_bins,
81-
output_dir=None,
82-
plot_figures=False,
83+
output_dir=output_dir,
84+
plot_figures=plot_figures,
8385
motion_bound=motion_bound,
8486
time_kernel_width=time_kernel_width,
8587
activity_network_hidden_features=activity_network_hidden_features,

0 commit comments

Comments
 (0)