mark test_async_completion_azure_caching as flaky

jwillians · Sep 21, 2024 · 16176da · 16176da
1 parent 5f1c1aa
commit 16176da
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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(