fix commits

alteryx · Nov 23, 2020 · 105169b · 105169b
1 parent e0f65c1
commit 105169b
Show file tree

Hide file tree

Showing 7 changed files with 12 additions and 5 deletions.
diff --git a/docs/source/release_notes.rst b/docs/source/release_notes.rst
@@ -17,6 +17,7 @@ Release Notes
         * Added a ``TimeSeriesRegressionPipeline`` class :pr:`1418`
         * Removed ``core-requirements.txt`` from the package distribution :pr:`1429`
         * Added a ``TimeSeriesSplit`` data splitter for time series problems :pr:`1441`.
+        * Added `FeatureTools Component` into transformer components :pr:`1454`
     * Fixes
         * Fixed ``IndexError`` raised in ``AutoMLSearch`` when ``ensembling = True`` but only one pipeline to iterate over :pr:`1397`
         * Fixed stacked ensemble input bug and LightGBM warning and bug in ``AutoMLSearch`` :pr:`1388`

diff --git a/evalml/pipelines/__init__.py b/evalml/pipelines/__init__.py
@@ -26,7 +26,8 @@
     DecisionTreeRegressor,
     StackedEnsembleClassifier,
     StackedEnsembleRegressor,
-    DelayedFeatureTransformer
+    DelayedFeatureTransformer,
+    FeatureTools
 )
 
 from .pipeline_base import PipelineBase

diff --git a/evalml/pipelines/components/__init__.py b/evalml/pipelines/components/__init__.py
@@ -37,7 +37,8 @@
     SelectColumns,
     TextFeaturizer,
     LSA,
-    PCA
+    PCA,
+    FeatureTools
 )
 from .ensemble import (
     StackedEnsembleClassifier,

diff --git a/evalml/pipelines/components/transformers/__init__.py b/evalml/pipelines/components/transformers/__init__.py
@@ -5,4 +5,4 @@
 from .scalers import StandardScaler
 from .column_selectors import DropColumns, SelectColumns
 from .dimensionality_reduction import PCA
-from .preprocessing import DateTimeFeaturizer, DropNullColumns, LSA, TextFeaturizer, DelayedFeatureTransformer
+from .preprocessing import DateTimeFeaturizer, DropNullColumns, LSA, TextFeaturizer, DelayedFeatureTransformer, FeatureTools
diff --git a/evalml/pipelines/components/transformers/preprocessing/__init__.py b/evalml/pipelines/components/transformers/preprocessing/__init__.py
@@ -4,3 +4,4 @@
 from .lsa import LSA
 from .text_featurizer import TextFeaturizer
 from .time_series import DelayedFeatureTransformer
+from .featuretools import FeatureTools
diff --git a/evalml/tests/component_tests/test_components.py b/evalml/tests/component_tests/test_components.py
@@ -33,6 +33,7 @@
     Estimator,
     ExtraTreesClassifier,
     ExtraTreesRegressor,
+    FeatureTools,
     Imputer,
     LightGBMClassifier,
     LinearRegressor,
@@ -146,6 +147,7 @@ def test_describe_component():
     text_featurizer = TextFeaturizer()
     lsa = LSA()
     pca = PCA()
+    ft = FeatureTools()
     assert enc.describe(return_dict=True) == {'name': 'One Hot Encoder', 'parameters': {'top_n': 10,
                                                                                         'features_to_encode': None,
                                                                                         'categories': None,
@@ -167,6 +169,7 @@ def test_describe_component():
     assert text_featurizer.describe(return_dict=True) == {'name': 'Text Featurization Component', 'parameters': {'text_columns': None}}
     assert lsa.describe(return_dict=True) == {'name': 'LSA Transformer', 'parameters': {'text_columns': None}}
     assert pca.describe(return_dict=True) == {'name': 'PCA Transformer', 'parameters': {'n_components': None, 'variance': 0.95}}
+    assert ft.describe(return_dict=True) == {'name': 'Featuretools Component', 'parameters': {}}
 
     # testing estimators
     base_classifier = BaselineClassifier()

diff --git a/evalml/tests/component_tests/test_utils.py b/evalml/tests/component_tests/test_utils.py
@@ -18,9 +18,9 @@
 
 def test_all_components(has_minimal_dependencies):
     if has_minimal_dependencies:
-        assert len(all_components()) == 29
+        assert len(all_components()) == 30
     else:
-        assert len(all_components()) == 35
+        assert len(all_components()) == 36
 
 
 def test_handle_component_class_names():