From fe1a6b5e9ee5ac251b28db019c84a763c743e345 Mon Sep 17 00:00:00 2001 From: Keita Onabuta Date: Wed, 27 Aug 2025 22:31:32 +0900 Subject: [PATCH] update loc second args - column 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 15eac2a8e8..939b10d757 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 (