diff --git a/tests/litellm/litellm_core_utils/test_streaming_handler.py b/tests/litellm/litellm_core_utils/test_streaming_handler.py index d79be260d8..7ee667d819 100644 --- a/tests/litellm/litellm_core_utils/test_streaming_handler.py +++ b/tests/litellm/litellm_core_utils/test_streaming_handler.py @@ -502,6 +502,7 @@ async def test_streaming_handler_with_usage( @pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.asyncio +@pytest.mark.flaky(reruns=3) async def test_streaming_with_usage_and_logging(sync_mode: bool): import time