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 (