From 16176da9fa2c61b1798ef405d7add7d06cf2cfaf Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 21 Sep 2024 11:41:25 -0700 Subject: [PATCH] mark test_async_completion_azure_caching as flaky --- litellm/tests/test_custom_callback_router.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_custom_callback_router.py b/litellm/tests/test_custom_callback_router.py index 57e1f264b..088d988e2 100644 --- a/litellm/tests/test_custom_callback_router.py +++ b/litellm/tests/test_custom_callback_router.py @@ -586,6 +586,7 @@ async def test_async_chat_azure_with_fallbacks(): # CACHING ## Test Azure - completion, embedding @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_async_completion_azure_caching(): customHandler_caching = CompletionCustomHandler() litellm.cache = Cache(