Fixing after rebase.

Author: JoeZiminski

src/spikeinterface/core/motion.py

@@ -67,7 +67,13 @@ def __repr__(self):
         else:
             rigid_txt = f"non-rigid - {nbins} spatial bins"
 
-        interval_s = self.temporal_bins_s[0][1] - self.temporal_bins_s[0][0]
+        if self.temporal_bins_s[0].size > 1:
+            interval_s = self.temporal_bins_s[0][1] - self.temporal_bins_s[0][0]
+        else:
+            # If there is only one temporal bin (entire session), assume the bin
+            # left edge is zero, and take twice it for the bin size.
+            interval_s = self.temporal_bins_s[0][0] * 2
+
         txt = f"Motion {rigid_txt} - interval {interval_s}s - {self.num_segments} segments"
         return txt
 
@@ -149,6 +155,12 @@ def get_displacement_at_time_and_depth(self, times_s, locations_um, segment_inde
         # reshape to grid domain shape if necessary
         displacement = displacement.reshape(out_shape)
 
+        # TODO: hacky
+        if self.temporal_bins_s[segment_index].size == 1 and self.spatial_bins_um.size == 1:
+            assert np.all(np.isnan(displacement))
+            assert self.displacement[segment_index].size == 1
+            displacement[:] = self.displacement[segment_index]
+
         return displacement
 
     def to_dict(self):

src/spikeinterface/sortingcomponents/motion/motion_interpolation.py

@@ -419,14 +419,10 @@ def __init__(
                 interpolation_time_bin_centers_s = motion.temporal_bins_s
                 interpolation_time_bin_edges_s = motion.temporal_bin_edges_s
         else:
-            interpolation_time_bin_centers_s, interpolation_time_bin_edges_s = (
-                ensure_time_bins(  # TODO: something is very wrong here with the typing
-                    interpolation_time_bin_centers_s, interpolation_time_bin_edges_s
-                )
+            interpolation_time_bin_centers_s, interpolation_time_bin_edges_s = ensure_time_bins(
+                interpolation_time_bin_centers_s, interpolation_time_bin_edges_s
             )
 
-        assert len(recording._recording_segments) == 1, "multi segment not supported"  # ??
-
         for segment_index, parent_segment in enumerate(recording._recording_segments):
             # finish the per-segment part of the time bin logic
             if interpolation_time_bin_centers_s is None:
@@ -440,8 +436,8 @@ def __init__(
                 )
                 assert segment_interpolation_time_bin_edges_s.shape == (segment_interpolation_time_bins_s.shape[0] + 1,)
             else:
-                segment_interpolation_time_bins_s = interpolation_time_bin_centers_s  # [segment_index]
-                segment_interpolation_time_bin_edges_s = interpolation_time_bin_edges_s  # [segment_index]
+                segment_interpolation_time_bins_s = interpolation_time_bin_centers_s[segment_index]
+                segment_interpolation_time_bin_edges_s = interpolation_time_bin_edges_s[segment_index]
 
             rec_segment = InterpolateMotionRecordingSegment(
                 parent_segment,

src/spikeinterface/sortingcomponents/motion/motion_utils.py

@@ -86,8 +86,6 @@ def get_spatial_windows(
         window_centers = np.arange(num_windows + 1) * win_step_um + min_ + border
         windows = []
 
-        print("CENTERS: ", window_centers.size)
-
         for win_center in window_centers:
             if win_shape == "gaussian":
                 win = np.exp(-((spatial_bin_centers - win_center) ** 2) / (2 * win_scale_um**2))
@@ -253,7 +251,7 @@ def make_2d_motion_histogram(
     arr[:, 1] = peak_locations[direction]
 
     if weight_with_amplitude:
-        weights = np.abs(peaks["amplitude"]) * 10
+        weights = np.abs(peaks["amplitude"])
     else:
         weights = None