From e33d634e5d216bd05c6ebc060edbae6ad4e7fc59 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 3 Sep 2024 11:56:03 -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 1b155bea850..4cc259b41f5 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -3712,6 +3712,7 @@ def test_unit_test_custom_stream_wrapper_function_call(): "vertex_ai/claude-3-5-sonnet@20240620", ], ) +@pytest.mark.flaky(retries=3, delay=1) def test_streaming_tool_calls_valid_json_str(model): if "vertex_ai" in model: from litellm.tests.test_amazing_vertex_completion import (