Fixed Dateframe to DataFrame in helpers (#1453)

LBHackney-IT · Oct 11, 2023 · 6d9b5bc · 6d9b5bc
1 parent c502832
commit 6d9b5bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/helpers/time_series_helpers.py b/scripts/helpers/time_series_helpers.py
@@ -336,7 +336,7 @@ def get_start_end_date(dataframe: pd.DataFrame, period: str, forecast_count: int
 
 
 def forecast_ets(dataframe: pd.DataFrame, start_date: str, end_date: str, seasonal_period: str = "none",
-                 error: str = "add", trend: str = "add", damped_trend: bool = False) -> pd.DateFrame:
+                 error: str = "add", trend: str = "add", damped_trend: bool = False) -> pd.DataFrame:
     """
         Args:
             Dataframe (dataframe): Dataframe containing training timeseries dataset.
@@ -381,7 +381,7 @@ def forecast_ets(dataframe: pd.DataFrame, start_date: str, end_date: str, season
 
 def forecast_holt_winters_ets(dataframe: pd.DataFrame, forecast_count: int, seasonal_period: str = "none",
                               trend: str = "add", use_boxcox: bool = False,
-                              initialization_method: str = "estimated") -> pd.DateFrame:
+                              initialization_method: str = "estimated") -> pd.DataFrame:
     """
 
             Args: