From 9c4e6d85ed80879cdbe5f5119ddd819531bb3fa9 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 12 Jun 2024 22:32:18 -0700 Subject: [PATCH] test: reduce vertex test to avoid exhausting limits --- litellm/tests/test_amazing_vertex_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 470be289c7..f624449aa0 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -503,7 +503,7 @@ async def test_async_vertexai_streaming_response(): # asyncio.run(test_async_vertexai_streaming_response()) -@pytest.mark.parametrize("provider", ["vertex_ai", "vertex_ai_beta"]) +@pytest.mark.parametrize("provider", ["vertex_ai"]) # "vertex_ai_beta" @pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.asyncio async def test_gemini_pro_vision(provider, sync_mode):