comments_jose

Nixtla · Dec 12, 2024 · 17951a7 · 17951a7
1 parent 6171f70
commit 17951a7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hierarchicalforecast/evaluation.py b/hierarchicalforecast/evaluation.py
@@ -495,11 +495,11 @@ def evaluate(
 
 # %% ../nbs/src/evaluation.ipynb 20
 def evaluate(
-    df: Frame,
+    df: FrameT,
     metrics: list[Callable],
     tags: dict[str, np.ndarray],
     models: Optional[list[str]] = None,
-    train_df: Optional[Frame] = None,
+    train_df: Optional[FrameT] = None,
     level: Optional[list[int]] = None,
     id_col: str = "unique_id",
     time_col: str = "ds",

diff --git a/nbs/src/evaluation.ipynb b/nbs/src/evaluation.ipynb
@@ -609,11 +609,11 @@
    "source": [
     "#| exporti\n",
     "def evaluate(\n",
-    "    df: Frame,\n",
+    "    df: FrameT,\n",
     "    metrics: list[Callable],\n",
     "    tags: dict[str, np.ndarray],\n",
     "    models: Optional[list[str]] = None,\n",
-    "    train_df: Optional[Frame] = None,\n",
+    "    train_df: Optional[FrameT] = None,\n",
     "    level: Optional[list[int]] = None,\n",
     "    id_col: str = \"unique_id\",\n",
     "    time_col: str = \"ds\",\n",