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 Nov 27, 2024
1 parent 2cff62b commit 22882ef
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/spikeinterface/benchmark/benchmark_motion_estimation.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,6 @@ def run(self, **job_kwargs):
estimate_motion=t4 - t3,
)


self.result["peaks"] = peaks
self.result["peak_locations"] = peak_locations
self.result["step_run_times"] = step_run_times
Expand Down Expand Up @@ -166,7 +165,9 @@ def create_benchmark(self, key):
def plot_true_drift(self, case_keys=None, scaling_probe=1.5, figsize=(8, 6)):
self.plot_drift(case_keys=case_keys, tested_drift=False, scaling_probe=scaling_probe, figsize=figsize)

def plot_drift(self, case_keys=None, gt_drift=True, tested_drift=True, raster=False, scaling_probe=1.0, figsize=(8, 6)):
def plot_drift(
self, case_keys=None, gt_drift=True, tested_drift=True, raster=False, scaling_probe=1.0, figsize=(8, 6)
):
import matplotlib.pyplot as plt

if case_keys is None:
Expand Down Expand Up @@ -206,7 +207,7 @@ def plot_drift(self, case_keys=None, gt_drift=True, tested_drift=True, raster=Fa
rec = bench.recording
x = peaks["sample_index"] / rec.sampling_frequency
y = peak_locations[bench.direction]
ax.scatter(x, y, alpha=.2, s=2, c=np.abs(peaks["amplitude"]), cmap="inferno")
ax.scatter(x, y, alpha=0.2, s=2, c=np.abs(peaks["amplitude"]), cmap="inferno")

for i in range(gt_motion.displacement[0].shape[1]):
depth = motion.spatial_bins_um[i]
Expand Down

0 comments on commit 22882ef

Please sign in to comment.