From e19107407bfbe94ace966c4ca60b605d4cb5eae7 Mon Sep 17 00:00:00 2001 From: Keita Onabuta Date: Sat, 30 Aug 2025 12:07:19 +0900 Subject: [PATCH] update loc second args - column (#1458) Configure second args of loc function to time_col instead of dataframe - X. --- flaml/automl/task/time_series_task.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flaml/automl/task/time_series_task.py b/flaml/automl/task/time_series_task.py index 15eac2a8e..939b10d75 100644 --- a/flaml/automl/task/time_series_task.py +++ b/flaml/automl/task/time_series_task.py @@ -529,7 +529,7 @@ def remove_ts_duplicates( duplicates = X.duplicated() if any(duplicates): - logger.warning("Duplicate timestamp values found in timestamp column. " f"\n{X.loc[duplicates, X][time_col]}") + logger.warning("Duplicate timestamp values found in timestamp column. " f"\n{X.loc[duplicates, time_col]}") X = X.drop_duplicates() logger.warning("Removed duplicate rows based on all columns") assert (