fix time calls

Nixtla · Dec 10, 2024 · b13053e · b13053e
1 parent b238b4d
commit b13053e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
@@ -2946,11 +2946,11 @@
     "        max_wait_time=max_wait_time,\n",
     "    )\n",
     "    mock_nixtla_client._make_request = side_effect\n",
-    "    init_time = time()\n",
+    "    init_time = time.time()\n",
     "    test_fail(\n",
     "        lambda: mock_nixtla_client.forecast(df=df, h=12, time_col='timestamp', target_col='value'),\n",
     "    )\n",
-    "    total_mock_time = time() - init_time\n",
+    "    total_mock_time = time.time() - init_time\n",
     "    if should_retry:\n",
     "        approx_expected_time = min((max_retries - 1) * retry_interval, max_wait_time)\n",
     "        upper_expected_time = min(max_retries * retry_interval, max_wait_time)\n",