From 085e1751ad41955b283339d3a4f81d4dd569491e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 12 Sep 2024 09:29:37 -0700 Subject: [PATCH] mark test as flaky --- litellm/tests/test_streaming.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 7a56d0703ff..2c4190382bb 100644 --- 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