diff --git a/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py b/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py index 860cabeb..b6a888b6 100644 --- a/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py +++ b/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py @@ -71,8 +71,7 @@ def fit(self, X: pd.DataFrame, y: pd.Series, s: Optional[pd.Series]) -> None: n_estimators=self.n_estimators, random_state=self.seed, n_jobs=self.n_jobs ) - features = pd.concat([X, y], axis=1) - features = pd.get_dummies(features) + features = pd.get_dummies(X) target = s.copy() features_train, features_val, target_train, target_val = train_test_split(