check timestamp_column

Signed-off-by: Wesley M. Gifford <[email protected]>
ibm-granite · Mar 29, 2024 · a008fb3 · a008fb3
1 parent 67ff1b3
commit a008fb3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)