diff --git a/ibis_ml/steps/__init__.py b/ibis_ml/steps/__init__.py index c2e9383..9be548c 100644 --- a/ibis_ml/steps/__init__.py +++ b/ibis_ml/steps/__init__.py @@ -6,9 +6,9 @@ OneHotEncode, TargetEncode, ) -from ibis_ml.steps._feature_engineering import CreatePolynomialFeatures -from ibis_ml.steps._feature_selection import DropZeroVariance +from ibis_ml.steps._generate_features import CreatePolynomialFeatures from ibis_ml.steps._impute import FillNA, ImputeMean, ImputeMedian, ImputeMode +from ibis_ml.steps._select_features import DropZeroVariance from ibis_ml.steps._standardize import ScaleMinMax, ScaleStandard from ibis_ml.steps._temporal import ExpandDate, ExpandDateTime, ExpandTime diff --git a/ibis_ml/steps/_feature_engineering.py b/ibis_ml/steps/_generate_features.py similarity index 100% rename from ibis_ml/steps/_feature_engineering.py rename to ibis_ml/steps/_generate_features.py diff --git a/ibis_ml/steps/_feature_selection.py b/ibis_ml/steps/_select_features.py similarity index 100% rename from ibis_ml/steps/_feature_selection.py rename to ibis_ml/steps/_select_features.py diff --git a/tests/test_feature_engineering.py b/tests/test_generate_features.py similarity index 100% rename from tests/test_feature_engineering.py rename to tests/test_generate_features.py diff --git a/tests/test_feature_selection.py b/tests/test_select_features.py similarity index 100% rename from tests/test_feature_selection.py rename to tests/test_select_features.py