From 6395bd8d652878cb1f503ae3c6fd85faaf7bf950 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 4 Apr 2025 20:25:05 -0700 Subject: [PATCH] test: mark flaky test --- tests/litellm/litellm_core_utils/test_streaming_handler.py | 1 + 1 file changed, 1 insertion(+) 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