diff --git a/src/aequitas/flow/methods/preprocessing/label_flipping.py b/src/aequitas/flow/methods/preprocessing/label_flipping.py index e52120d7..d1427115 100644 --- a/src/aequitas/flow/methods/preprocessing/label_flipping.py +++ b/src/aequitas/flow/methods/preprocessing/label_flipping.py @@ -273,7 +273,7 @@ def _label_flipping(self, y: pd.Series, s: Optional[pd.Series], scores: pd.Serie if self.ordering_method == "residuals" else y_flipped.loc[scores <= 0].index ) - to_flip = pd.Series(index=flip_index).fillna(False).loc[y_flipped.index] + to_flip = pd.Series(index=flip_index).fillna(False) for group, flips in group_flips.items(): if flips > 0: labels = y == 0