From 2e40ab959d198aaa59ab69fe9c0c63291486142b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 4 Apr 2024 13:54:16 -0700 Subject: [PATCH] test(test_router.py): fix casting --- litellm/tests/test_router.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 82547e5a3..3a6043b5a 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -1005,7 +1005,7 @@ def test_reading_keys_os_environ(): assert client.max_retries == int( os.environ["AZURE_MAX_RETRIES"] ), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" - assert client.timeout == ( + assert client.timeout == int( os.environ["AZURE_TIMEOUT"] ), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("sync client set correctly!") @@ -1017,7 +1017,7 @@ def test_reading_keys_os_environ(): assert stream_client.max_retries == int( os.environ["AZURE_MAX_RETRIES"] ), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" - assert stream_client.timeout == ( + assert stream_client.timeout == int( os.environ["AZURE_STREAM_TIMEOUT"] ), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("sync stream client set correctly!") @@ -1102,7 +1102,7 @@ def test_reading_openai_keys_os_environ(): assert client.max_retries == int( os.environ["AZURE_MAX_RETRIES"] ), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" - assert client.timeout == ( + assert client.timeout == int( os.environ["AZURE_TIMEOUT"] ), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("sync client set correctly!") @@ -1113,7 +1113,7 @@ def test_reading_openai_keys_os_environ(): assert stream_client.max_retries == int( os.environ["AZURE_MAX_RETRIES"] ), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" - assert stream_client.timeout == ( + assert stream_client.timeout == int( os.environ["AZURE_STREAM_TIMEOUT"] ), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("sync stream client set correctly!")