Fix tests

alteryx · Mar 2, 2023 · 9076320 · 9076320
1 parent 19dd533
commit 9076320
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/evalml/tests/integration_tests/test_data_checks_and_actions_integration.py b/evalml/tests/integration_tests/test_data_checks_and_actions_integration.py
@@ -20,7 +20,11 @@
     TimeSeriesRegressionPipeline,
     TimeSeriesRegularizer,
 )
-from evalml.pipelines.components import DropColumns, DropRowsTransformer, TargetImputer
+from evalml.pipelines.components import (
+    DropColumns,
+    DropRowsTransformer,
+    TargetImputer,
+)
 from evalml.pipelines.components.transformers.imputers.per_column_imputer import (
     PerColumnImputer,
 )
@@ -218,7 +222,7 @@ def test_data_checks_impute_cols(problem_type):
         expected_pipeline_class = BinaryClassificationPipeline
         y_expected = ww.init_series(
             pd.Series([0, 1, 1, 1, 1]),
-            logical_type="Integer",
+            logical_type="Double",
         )
 
     elif problem_type == "multiclass":
@@ -227,7 +231,7 @@ def test_data_checks_impute_cols(problem_type):
         expected_pipeline_class = MulticlassClassificationPipeline
         y_expected = ww.init_series(
             pd.Series([0, 1, 2, 2, 2]),
-            logical_type="Integer",
+            logical_type="Double",
         )
 
     else: