compare

wenfeiy-db · Jan 18, 2024 · 3751d79 · 3751d79
1 parent caa014c
commit 3751d79
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/runtime/databricks/automl_runtime/forecast/utils.py b/runtime/databricks/automl_runtime/forecast/utils.py
@@ -111,6 +111,7 @@ def get_validation_horizon(df: pd.DataFrame, horizon: int, unit: str) -> int:
     horizon_dateoffset = pd.DateOffset(**DATE_OFFSET_KEYWORD_MAP[unit]) * horizon
 
     try:
+        df["ds"] = pd.to_datetime(df["ds"])
         if MIN_HORIZONS * horizon_dateoffset + df["ds"].min() <= df["ds"].max():
             return horizon
     except OverflowError: