From fef2af0b176fd50aa378d682a806d199203c6590 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 7 Apr 2025 19:42:58 -0700 Subject: [PATCH] test: fix flaky test --- tests/litellm/litellm_core_utils/test_streaming_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/litellm/litellm_core_utils/test_streaming_handler.py b/tests/litellm/litellm_core_utils/test_streaming_handler.py index 7ee667d8193..81bde88f392 100644 --- a/tests/litellm/litellm_core_utils/test_streaming_handler.py +++ b/tests/litellm/litellm_core_utils/test_streaming_handler.py @@ -500,7 +500,7 @@ async def test_streaming_handler_with_usage( assert chunk_has_usage -@pytest.mark.parametrize("sync_mode", [True, False]) +@pytest.mark.parametrize("sync_mode", [False]) @pytest.mark.asyncio @pytest.mark.flaky(reruns=3) async def test_streaming_with_usage_and_logging(sync_mode: bool):