mark test as flaky

dbczumar · Sep 12, 2024 · 085e175 · 085e175
1 parent bea34c9
commit 085e175
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py
@@ -1431,6 +1431,7 @@ async def test_completion_replicate_llama3_streaming(sync_mode):
     ],
 )
 @pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_bedrock_httpx_streaming(sync_mode, model, region):
     try:
         litellm.set_verbose = True