use assert_frame_equal in df checks

Nixtla · Mar 25, 2024 · f20053e · f20053e
1 parent 5a4ee44
commit f20053e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nbs/timegpt.ipynb b/nbs/timegpt.ipynb
@@ -1683,11 +1683,11 @@
     "test_warns(\n",
     "    lambda: timegpt.forecast(df_test, finetune_steps=2, h=12, model=\"timegpt-1-long-horizon\"),\n",
     ")\n",
-    "test_eq(\n",
+    "pd.testing.assert_frame_equal(\n",
     "    timegpt.forecast(df_test, h=12, model=\"timegpt-1\"),\n",
     "    timegpt.forecast(df_test, h=12),\n",
     ")\n",
-    "test_eq(\n",
+    "pd.testing.assert_frame_equal(\n",
     "    timegpt.forecast(df_test, h=12, model=\"timegpt-1-long-horizon\"),\n",
     "    timegpt.forecast(df_test, h=12, model=\"long-horizon\"),\n",
     ")"