From 6d9b5bc0629028a50482fdeae17122b597808ae3 Mon Sep 17 00:00:00 2001
From: HuuHieuDo <77067292+HuuHieuDo@users.noreply.github.com>
Date: Wed, 11 Oct 2023 16:01:42 +0100
Subject: [PATCH] Fixed Dateframe to DataFrame in helpers (#1453)

---
 scripts/helpers/time_series_helpers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/helpers/time_series_helpers.py b/scripts/helpers/time_series_helpers.py
index 63b49e0ea..36519cc0c 100644
--- 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: