Added support for ndarrays for featurizer

alteryx · Jan 31, 2024 · 701eef6 · 701eef6
1 parent e49dfa5
commit 701eef6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/evalml/pipelines/components/transformers/preprocessing/time_series_featurizer.py b/evalml/pipelines/components/transformers/preprocessing/time_series_featurizer.py
@@ -135,8 +135,9 @@ def fit(self, X, y=None):
             )
         else:
             # For the multiseries case, each series ID has individualized lag values
-            if isinstance(y, pd.Series):
-                y = y.to_frame()
+            if isinstance(y, pd.Series) or isinstance(y, np.ndarray):
+                y = pd.DataFrame(y)
+
             self.statistically_significant_lags = {}
             for column in y.columns:
                 self.statistically_significant_lags[