Fix lint

alteryx · Nov 15, 2021 · 731a3f4 · 731a3f4
1 parent 4d85eee
commit 731a3f4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/evalml/pipelines/components/transformers/preprocessing/delayed_feature_transformer.py b/evalml/pipelines/components/transformers/preprocessing/delayed_feature_transformer.py
@@ -171,14 +171,12 @@ def transform(self, X, y=None):
         If y is not None, it will also compute the delayed values for the target variable.
 
         Args:
-            X (pd.DataFrame or None): Data to transform. None is expected when only the target variable is being used.
+            X (pd.DataFrame): Data to transform. None is expected when only the target variable is being used.
             y (pd.Series, or None): Target.
 
         Returns:
             pd.DataFrame: Transformed X.
         """
-        if X is None:
-            X = pd.DataFrame()
         # Normalize the data into pandas objects
         X_ww = infer_feature_types(X)
         cols_to_delay = list(
@@ -213,7 +211,7 @@ def fit_transform(self, X, y):
         """Fit the component and transform the input data.
 
         Args:
-            X (pd.DataFrame or None): Data to transform. None is expected when only the target variable is being used.
+            X (pd.DataFrame): Data to transform. None is expected when only the target variable is being used.
             y (pd.Series, or None): Target.
 
         Returns:

diff --git a/evalml/tests/automl_tests/test_automl_utils.py b/evalml/tests/automl_tests/test_automl_utils.py
@@ -112,7 +112,7 @@ def test_make_data_splitter_default(problem_type, large_data):
         assert data_splitter.n_splits == 3
         assert data_splitter.gap == 1
         assert data_splitter.max_delay == 7
-        assert data_splitter.date_index is "foo"
+        assert data_splitter.date_index == "foo"
 
 
 @pytest.mark.parametrize(
@@ -160,7 +160,7 @@ def test_make_data_splitter_parameters_time_series():
         assert data_splitter.n_splits == 5
         assert data_splitter.gap == 1
         assert data_splitter.max_delay == 7
-        assert data_splitter.date_index is "foo"
+        assert data_splitter.date_index == "foo"
 
 
 def test_make_data_splitter_error():