diff --git a/runtime/databricks/automl_runtime/forecast/pmdarima/training.py b/runtime/databricks/automl_runtime/forecast/pmdarima/training.py index a3f6ba8a..88233961 100644 --- a/runtime/databricks/automl_runtime/forecast/pmdarima/training.py +++ b/runtime/databricks/automl_runtime/forecast/pmdarima/training.py @@ -138,7 +138,7 @@ def _fit_predict(self, df: pd.DataFrame, cutoffs: List[pd.Timestamp], seasonal_p @staticmethod def _fill_missing_time_steps(df: pd.DataFrame, frequency: str): # Forward fill missing time steps - df_filled = df.set_index("ds").resample(rule=OFFSET_ALIAS_MAP[frequency]).pad().reset_index() + df_filled = df.set_index("ds").resample(rule=OFFSET_ALIAS_MAP[frequency]).ffill().reset_index() start_ds, modified_start_ds = df["ds"].min(), df_filled["ds"].min() if start_ds != modified_start_ds: offset = modified_start_ds - start_ds