diff --git a/imblearn/under_sampling/condensed_nearest_neighbour.py b/imblearn/under_sampling/condensed_nearest_neighbour.py index 3656dff13..b4030c0a2 100644 --- a/imblearn/under_sampling/condensed_nearest_neighbour.py +++ b/imblearn/under_sampling/condensed_nearest_neighbour.py @@ -244,7 +244,7 @@ def _sample(self, X, y): np.flatnonzero(pred_S_y == S_y))) # Find the misclassified S_y - sel_x = np.squeeze(S_x[idx_maj_sample, :]) + sel_x = S_x[idx_maj_sample, :] sel_y = S_y[idx_maj_sample] # The indexes found are relative to the current class, we need to