test different hist exogs

Nixtla · Nov 13, 2024 · 91b27ec · 91b27ec
1 parent 19c80f5
commit 91b27ec
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
@@ -2292,6 +2292,27 @@
     "    )"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| hide\n",
+    "# different hist exog, different results\n",
+    "for X_df in (None, valid):\n",
+    "    res1 = nixtla_client.forecast(train, h=h, X_df=X_df, freq=freq, hist_exog_list=x_cols[:2])\n",
+    "    res2 = nixtla_client.forecast(train, h=h, X_df=X_df, freq=freq, hist_exog_list=x_cols[2:])\n",
+    "    np.testing.assert_raises(\n",
+    "        AssertionError,\n",
+    "        np.testing.assert_allclose,\n",
+    "        res1['TimeGPT'],\n",
+    "        res2['TimeGPT'],\n",
+    "        atol=1e-4,\n",
+    "        rtol=1e-3,\n",
+    "    )"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,