From 9a3873b9edcfcb8477a3f05504ed38dfac329035 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 31 Aug 2024 15:02:56 -0700 Subject: [PATCH] mark flaky test as flaky --- litellm/tests/test_custom_callback_input.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index 739365a702e..1d53682923e 100644 --- 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()