add sleep

Nixtla · Dec 10, 2024 · b238b4d · b238b4d
1 parent 1c84c21
commit b238b4d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
@@ -140,9 +140,9 @@
    "outputs": [],
    "source": [
     "#| hide\n",
+    "import time\n",
     "from contextlib import contextmanager\n",
     "from itertools import product\n",
-    "from time import time, sleep\n",
     "\n",
     "from dotenv import load_dotenv\n",
     "from fastcore.test import test_eq, test_fail\n",
@@ -2459,9 +2459,10 @@
     "model_id2 = nixtla_client.finetune(train, finetuned_model_id=model_id1)\n",
     "\n",
     "# list finetuned models\n",
+    "time.sleep(2)  # it can take a bit for the models to show up\n",
     "models = nixtla_client.finetuned_models()\n",
     "ids = {m.id for m in models}\n",
-    "assert {model_id1, model_id2} <= ids, [{model_id1, model_id2}, ids]"
+    "assert {model_id1, model_id2} <= ids"
    ]
   },
   {
@@ -3014,7 +3015,7 @@
     "def raise_read_timeout_error(*args, **kwargs):\n",
     "    sleep_seconds = 5\n",
     "    print(f'raising ReadTimeout error after {sleep_seconds} seconds')\n",
-    "    sleep(sleep_seconds)\n",
+    "    time.sleep(sleep_seconds)\n",
     "    raise httpx.ReadTimeout('Timed out')\n",
     "\n",
     "def raise_http_error(*args, **kwargs):\n",