From 1ed96b1fc89bcce8ec4f434d60de6975f1f1fc82 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 30 Dec 2023 15:54:46 +0530 Subject: [PATCH] test(test_router.py): add retries --- litellm/tests/test_router.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 987e6670f8..c79a838835 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -441,7 +441,7 @@ async def test_aimg_gen_on_router(): }, }, ] - router = Router(model_list=model_list) + router = Router(model_list=model_list, num_retries=3) response = await router.aimage_generation( model="dall-e-3", prompt="A cute baby sea otter" )