Moved feat type infer

alteryx · Oct 26, 2023 · 5b7f213 · 5b7f213
1 parent 2f4ad3c
commit 5b7f213
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/evalml/pipelines/multiseries_regression_pipeline.py b/evalml/pipelines/multiseries_regression_pipeline.py
@@ -6,6 +6,7 @@
     TimeSeriesRegressionPipeline,
 )
 from evalml.problem_types import ProblemTypes
+from evalml.utils import infer_feature_types
 
 
 class MultiseriesRegressionPipeline(TimeSeriesRegressionPipeline):
@@ -129,8 +130,15 @@ def predict_in_sample(
         )
         # Order series ID columns to be same as X_train
         y_unstacked = y_unstacked[y_train_unstacked.columns]
+
+        X_train_unstacked = infer_feature_types(X_train_unstacked)
+        y_train_unstacked = infer_feature_types(y_train_unstacked)
+        X_unstacked = infer_feature_types(X_unstacked)
+        y_unstacked = infer_feature_types(y_unstacked)
+
         print("inuns_X", X_unstacked, X_unstacked.ww.schema)
         print("inuns_X_train", X_train_unstacked, X_train_unstacked.ww.schema)
+
         unstacked_predictions = super().predict_in_sample(
             X_unstacked,
             y_unstacked,

diff --git a/evalml/pipelines/utils.py b/evalml/pipelines/utils.py
@@ -1435,10 +1435,6 @@ def unstack_multiseries(
     # Reset the axes now that they've been unstacked, keep time info in X
     X_unstacked = X_unstacked.reset_index()
     y_unstacked = y_unstacked.reset_index(drop=True)
-
-    X_unstacked = infer_feature_types(X_unstacked)
-    y_unstacked = infer_feature_types(y_unstacked)
-
     return X_unstacked, y_unstacked
 
 
@@ -1495,8 +1491,6 @@ def stack_data(
         )
         stacked_series = pd.concat([series_id_col, stacked_series], axis=1)
 
-    stacked_series = infer_feature_types(stacked_series)
-
     return stacked_series