fix object path

VIDA-NYU · Jun 2, 2024 · 624dac2 · 624dac2
1 parent 96ce4a1
commit 624dac2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alpha_automl/resource/primitives_hierarchy.json b/alpha_automl/resource/primitives_hierarchy.json
@@ -35,8 +35,8 @@
         "sklearn.feature_selection.GenericUnivariateSelect",
         "sklearn.feature_selection.SelectPercentile",
         "sklearn.feature_selection.SelectKBest",
-        "feature_engine.selection.SmartCorrelatedSelection",
-        "feature_engine.selection.DropHighPSIFeatures"
+        "feature_engine.selection.smart_correlation_selection.SmartCorrelatedSelection",
+        "feature_engine.selection.drop_psi_features.DropHighPSIFeatures"
     ],
     "COLUMN_TRANSFORMER": [
         "sklearn.compose.ColumnTransformer"