diff --git a/evalml/tests/automl_tests/test_automl.py b/evalml/tests/automl_tests/test_automl.py index 24a5bf5b79..d26ad9ba02 100644 --- a/evalml/tests/automl_tests/test_automl.py +++ b/evalml/tests/automl_tests/test_automl.py @@ -70,6 +70,7 @@ ARIMARegressor, DateTimeFeaturizer, EmailFeaturizer, + LinearRegressor, NaturalLanguageFeaturizer, RandomForestClassifier, SelectColumns, @@ -5377,6 +5378,7 @@ def test_excluded_model_families( assert SelectColumns.name not in pl.component_graph.compute_order else: assert RandomForestClassifier.name not in pl.component_graph.compute_order + assert LinearRegressor.name not in pl.component_graph.compute_order def test_excluded_model_families_error(