mark flaky test as flaky

BerriAI · Aug 31, 2024 · 9a3873b · 9a3873b
1 parent 2e0ee8c
commit 9a3873b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py
@@ -976,6 +976,7 @@ async def test_async_embedding_bedrock():
 # CACHING
 ## Test Azure - completion, embedding
 @pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_async_completion_azure_caching():
     litellm.set_verbose = True
     customHandler_caching = CompletionCustomHandler()