fix keep_list

zm711 · Nov 27, 2023 · ff7c5b1 · ff7c5b1
1 parent bbd4175
commit ff7c5b1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -357,7 +357,6 @@ def plot_raster(
         window: Union[list, list[list]],
         show_stim: bool = True,
         include_ids: list | np.nadarry | None = None,
-        sorted: bool = False,
     ):
         """
         Function to plot rasters
@@ -397,7 +396,7 @@ def plot_raster(
                 keep_list.append(np.where(cluster_indices == cid)[0][0])
             keep_list = np.array(keep_list)
         else:
-            keep_list = np.arange(0, len(cluster_indices), 1)
+            keep_list = np.arange(0, len(self.data.cluster_ids), 1)
 
         for idx, stimulus in enumerate(psths.keys()):
             bins = psths[stimulus]["bins"]