From c7a97e7bb312b318d3a3dcec59b6a25a464daabc Mon Sep 17 00:00:00 2001 From: Becca McBrayer Date: Wed, 26 Jul 2023 15:05:47 -0400 Subject: [PATCH] Update evalml/pipelines/utils.py Co-authored-by: chukarsten <64713315+chukarsten@users.noreply.github.com> --- evalml/pipelines/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/evalml/pipelines/utils.py b/evalml/pipelines/utils.py index 54cf11ac15..bc5baa33ce 100644 --- a/evalml/pipelines/utils.py +++ b/evalml/pipelines/utils.py @@ -1385,9 +1385,7 @@ def unstack_multiseries(X, y, series_id, time_index, keep_time_in_index=True): # Save the time_index for alignment new_time_index = single_series[time_index] - for column_name in full_dataset.columns: - if column_name in [time_index, series_id]: - continue + for column_name in full_dataset.columns.drop([time_index, series_id]): new_column = single_series[column_name] new_column.index = new_time_index