handle pandas<2.2 alias

Nixtla · Dec 2, 2024 · 39a36e6 · 39a36e6
1 parent 49685fe
commit 39a36e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
@@ -2383,7 +2383,7 @@
     "client.cross_validation(df=series, freq=custom_business_hours, h=7)\n",
     "fcst = client.forecast(df=series, freq=custom_business_hours, h=7)\n",
     "assert sorted(fcst['ds'].dt.hour.unique().tolist()) == list(range(9, 16))\n",
-    "assert list(client._model_params.keys()) == [('timegpt-1', 'cbh')]"
+    "assert [(model, freq.lower()) for (model, freq) in client._model_params.keys()] == [('timegpt-1', 'cbh')]"
    ]
   },
   {