fix?

wenfeiy-db · Jan 17, 2024 · 3de71ad · 3de71ad
1 parent c9ab5a0
commit 3de71ad
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/runtime/databricks/automl_runtime/forecast/prophet/model.py b/runtime/databricks/automl_runtime/forecast/prophet/model.py
@@ -239,7 +239,10 @@ def predict_timeseries(self, horizon: int = None, include_history: bool = True)
             identity_column_names=self._id_cols
         )
         future_df["ts_id"] = future_df[self._id_cols].apply(tuple, axis=1)
-        return future_df.groupby(self._id_cols).apply(lambda df: self._predict_impl(df, horizon, include_history)).reset_index()
+        print(f"predict_timeseries::future_df:\n {future_df}")
+        return future_df.groupby(self._id_cols).apply(
+                lambda df: self._predict_impl(df, horizon, include_history)
+            ).reset_index(drop=True)
 
     @staticmethod
     def get_reserved_cols() -> List[str]: