mark test as flaky

2024-08-28 08:24:04 -07:00 · 2024-08-28 08:24:04 -07:00 · b373320569
commit b373320569
parent 3cec00939e
1 changed files with 1 additions and 0 deletions
--- 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()