minor fixes

Author: jakeswann1

src/spikeinterface/widgets/amplitudes.py

@@ -83,7 +83,7 @@ def __init__(
         # For SortingView, ensure we're only using a single segment
         if is_sortingview and len(segment_indices) > 1:
             warn("SortingView backend currently supports only single segment. Using first segment.")
-            segment_indices = segment_indices[0]
+            segment_indices = [segment_indices[0]]
 
         # Create multi-segment data structure (dict of dicts)
         spiketrains_by_segment = {}
@@ -150,11 +150,13 @@ def __init__(
             first_segment = segment_indices[0]
             plot_data["spike_train_data"] = spiketrains_by_segment[first_segment]
             plot_data["y_axis_data"] = amplitudes_by_segment[first_segment]
+            print(plot_data["spike_train_data"])
+            print(plot_data["y_axis_data"])
         else:
             # Otherwise use the full dict of dicts structure with all segments
             plot_data["spike_train_data"] = spiketrains_by_segment
             plot_data["y_axis_data"] = amplitudes_by_segment
-            plot_data["segment_index"] = segment_indices
+            plot_data["segment_indices"] = segment_indices
 
         BaseRasterWidget.__init__(self, **plot_data, backend=backend, **backend_kwargs)
 

src/spikeinterface/widgets/motion.py

@@ -201,10 +201,10 @@ def __init__(
             if len(unique_segments) == 1:
                 segment_indices = [int(unique_segments[0])]
             else:
-                raise ValueError("segment_index must be specified if there are multiple segments")
+                raise ValueError("segment_indices must be specified if there are multiple segments")
 
         if not isinstance(segment_indices, list):
-            raise ValueError("segment_index must be an int or a list of ints")
+            raise ValueError("segment_indices must be a list of ints")
 
         # Validate all segment indices exist in the data
         for idx in segment_indices:
@@ -275,7 +275,7 @@ def __init__(
         plot_data = dict(
             spike_train_data=spike_train_data,
             y_axis_data=y_axis_data,
-            segment_index=segment_indices,
+            segment_indices=segment_indices,
             y_lim=depth_lim,
             color_kwargs=color_kwargs,
             scatter_decimate=scatter_decimate,
@@ -417,7 +417,7 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
         commpon_drift_map_kwargs = dict(
             direction=dp.motion.direction,
             recording=dp.recording,
-            segment_index=dp.segment_index,
+            segment_indices=list(dp.segment_index),
             depth_lim=dp.depth_lim,
             scatter_decimate=dp.scatter_decimate,
             color_amplitude=dp.color_amplitude,

src/spikeinterface/widgets/rasters.py

@@ -419,7 +419,7 @@ def __init__(
 
         sorting = self.ensure_sorting(sorting)
 
-        segment_indices = validate_segment_indices(sorting, segment_indices)
+        segment_indices = validate_segment_indices(segment_indices, sorting)
 
         if unit_ids is None:
             unit_ids = sorting.unit_ids
@@ -479,7 +479,7 @@ def __init__(
         plot_data = dict(
             spike_train_data=spike_train_data,
             y_axis_data=y_axis_data,
-            segment_index=segment_indices,
+            segment_indices=segment_indices,
             x_lim=time_range,
             y_label="Unit id",
             unit_ids=unit_ids,