test: mark flaky test

2025-04-04 20:25:05 -07:00 · 2025-04-04 20:25:05 -07:00 · 6395bd8d65
commit 6395bd8d65
parent c555c15ad7
1 changed files with 1 additions and 0 deletions
--- 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