mark test_async_completion_azure_caching as flaky

2024-09-21 11:41:25 -07:00 · 2024-09-21 11:41:25 -07:00 · 16176da9fa
commit 16176da9fa
parent 5f1c1aaa04
1 changed files with 1 additions and 0 deletions
--- 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(