From 7d6f68c1295155765666a6bc05e3ef561b76fe5d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9rgio=20Jesus?= Date: Wed, 11 Sep 2024 11:21:45 +0100 Subject: [PATCH] Remove loc with non existing indexes --- src/aequitas/flow/methods/preprocessing/label_flipping.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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