Use feature names

alteryx · Oct 31, 2023 · 6de173a · 6de173a
1 parent aa7ca03
commit 6de173a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/evalml/pipelines/multiseries_regression_pipeline.py b/evalml/pipelines/multiseries_regression_pipeline.py
@@ -133,7 +133,9 @@ def predict_in_sample(
             y_train_unstacked.columns.intersection(y_unstacked.columns)
         ]
         X_unstacked = X_unstacked[
-            X_train_unstacked.columns.intersection(X_unstacked.columns)
+            X_train_unstacked.columns.intersection(
+                set(self.input_feature_names.values())[0],
+            )
         ]
 
         X_train_unstacked = infer_feature_types(X_train_unstacked)