conflict

SpikeInterface · Jan 8, 2025 · 0128d8f · 0128d8f
2 parents 9a17967 + 0190e88
commit 0128d8f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/spikeinterface/sortingcomponents/motion/medicine.py b/src/spikeinterface/sortingcomponents/motion/medicine.py
@@ -34,7 +34,6 @@ def run(
         progress_bar,
         extra,
         bin_s=1.0,
-
         ## medicine specific kwargs propagated to the lib
         motion_bound=800,
         time_kernel_width=30,
@@ -70,6 +69,7 @@ def run(
 
         if optimizer is None:
             import torch
+
             optimizer = torch.optim.Adam
 
         trainer, time_bins, depth_bins, pred_motion = run_medicine(
@@ -78,10 +78,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,
-
             motion_bound=motion_bound,
             time_kernel_width=time_kernel_width,
             activity_network_hidden_features=activity_network_hidden_features,