From b13053e04fa4344ec8b27e6aee3744c91559b14e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Morales?= Date: Tue, 10 Dec 2024 10:21:21 -0600 Subject: [PATCH] fix time calls --- nbs/src/nixtla_client.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb index d7daf1e5..fd0879f7 100644 --- 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",