Merge branch 'main' into feature/kan4ts

Nixtla · Sep 16, 2024 · 32662f7 · 32662f7
2 parents 4ab1032 + 42fa79e
commit 32662f7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -1234,7 +1234,10 @@
     "                    val_size=val_size,\n",
     "                    sort_df=sort_df,\n",
     "                    use_init_models=False,\n",
-    "                    verbose=verbose,                 \n",
+    "                    verbose=verbose,\n",
+    "                    id_col=id_col,\n",
+    "                    time_col=time_col,\n",
+    "                    target_col=target_col,                                     \n",
     "                )\n",
     "                predict_df: Optional[DataFrame] = None\n",
     "            else:\n",

diff --git a/neuralforecast/core.py b/neuralforecast/core.py
@@ -1193,6 +1193,9 @@ def cross_validation(
                     sort_df=sort_df,
                     use_init_models=False,
                     verbose=verbose,
+                    id_col=id_col,
+                    time_col=time_col,
+                    target_col=target_col,
                 )
                 predict_df: Optional[DataFrame] = None
             else: