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(