fix keep_list in sm_fr

zm711 · Nov 27, 2023 · 1a47ba0 · 1a47ba0
1 parent ff7c5b1
commit 1a47ba0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -528,7 +528,7 @@ def plot_sm_fr(
                 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)
 
         stim_trial_groups = self._get_trial_groups()
         event_lengths = self._get_event_lengths_all()