check for linear models

alteryx · Aug 24, 2023 · 4580a41 · 4580a41
1 parent 9bc0f44
commit 4580a41
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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(