fix: rm ex vars num partitions cv

Nixtla · Nov 16, 2023 · c5492a6 · c5492a6
1 parent 7a5f056
commit c5492a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/timegpt.ipynb b/nbs/timegpt.ipynb
@@ -1610,7 +1610,7 @@
     "        timegpt.cross_validation,\n",
     "        h=7,\n",
     "        n_windows=4,\n",
-    "        df=df_freq,\n",
+    "        df=df_freq.drop(columns='ex_1') if freq == 'D' else df_freq,\n",
     "        num_partitions=2,\n",
     "    )\n",
     "    test_num_partitions_same_results(\n",