From a25048940502f7021a1ddc2979ef27be98d8d008 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 9 Dec 2023 12:28:28 -0800 Subject: [PATCH] test(test_router.py): run as asyncio.run --- litellm/tests/test_router.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 6f43ed99f12..cd65aaf5613 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -226,12 +226,9 @@ def test_call_one_endpoint(): ) print("\n response", response) - tasks = [] - tasks.append(call_azure_completion()) - tasks.append(call_bedrock_claude()) - tasks.append(call_azure_embedding()) - - asyncio.gather(**tasks) + asyncio.run(call_azure_completion()) + asyncio.run(call_bedrock_claude()) + asyncio.run(call_azure_embedding() os.environ["AZURE_API_BASE"] = old_api_base os.environ["AZURE_API_KEY"] = old_api_key