From b3733205698f54fce86e93255d40a48232fddf6d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 28 Aug 2024 08:24:04 -0700 Subject: [PATCH] mark test as flaky --- litellm/tests/test_amazing_vertex_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 53005fac09..c967577be3 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -846,6 +846,7 @@ def test_gemini_pro_grounding(value_in_dict): ) # "vertex_ai", @pytest.mark.parametrize("sync_mode", [True]) # "vertex_ai", @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_gemini_pro_function_calling_httpx(model, sync_mode): try: load_vertex_ai_credentials()