diff --git a/tsfm_public/toolkit/time_series_preprocessor.py b/tsfm_public/toolkit/time_series_preprocessor.py index 369a9250..c0b9345d 100644 --- a/tsfm_public/toolkit/time_series_preprocessor.py +++ b/tsfm_public/toolkit/time_series_preprocessor.py @@ -127,6 +127,9 @@ def __init__( if not isinstance(id_columns, list): raise ValueError(f"Invalid argument provided for `id_columns`: {id_columns}") + if isinstance(timestamp_column, list): + raise ValueError(f"`timestamp_column` should not be a list, received: {timestamp_column}") + self.id_columns = id_columns self.timestamp_column = timestamp_column self.target_columns = list(target_columns)