Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 8, 2025
1 parent 133cb2a commit 0190e88
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/spikeinterface/sortingcomponents/motion/medecine.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@ def run(
progress_bar,
extra,
bin_s=1.0,

## medecine specific kwargs propagated to the lib
motion_bound=800,
time_kernel_width=30,
Expand Down Expand Up @@ -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(
Expand All @@ -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,
Expand Down

0 comments on commit 0190e88

Please sign in to comment.