pass -> continue

zm711 · Nov 7, 2023 · c35b6bc · c35b6bc
1 parent a847815
commit c35b6bc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -389,7 +389,7 @@ def plot_raster(self, window: Union[list, list[list]], show_stim: bool = True, i
         if sorted:
             sorted_indices = []
             for value_id in include_ids:
-                sorted_indices.append(np.nonzero(self.cluster_ids==value_id)[0][0])
+                sorted_indices.append(np.nonzero(self.data.cluster_ids==value_id)[0][0])
             sorted_indices = np.array(sorted_indices)
 
         for idx, stimulus in enumerate(psths.keys()):
@@ -410,7 +410,7 @@ def plot_raster(self, window: Union[list, list[list]], show_stim: bool = True, i
 
             for idy in range(np.shape(psth)[0]):
                 if idy not in keep_list:
-                    pass
+                    continue
                 psth_sub = np.squeeze(psth[idy])
 
                 if np.sum(psth_sub) == 0:
@@ -534,7 +534,7 @@ def plot_sm_fr(
         if sorted:
             sorted_indices = []
             for value_id in include_ids:
-                sorted_indices.append(np.nonzero(self.cluster_ids==value_id)[0][0])
+                sorted_indices.append(np.nonzero(self.data.cluster_ids==value_id)[0][0])
             sorted_indices = np.array(sorted_indices)
 
         stim_trial_groups = self._get_trial_groups()
@@ -571,7 +571,7 @@ def plot_sm_fr(
             event_len = np.zeros((len(tg_set)))
             for cluster_number in range(np.shape(psth)[0]):
                 if cluster_number not in keep_list:
-                    pass
+                    continue
                 smoothed_psth = gaussian_smoothing(psth[cluster_number], bin_size, sm_std)
 
                 for trial_number, trial in enumerate(tg_set):