From c5df662e0f071247a4d2b7e5b52af93c79b704cb Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 13 Jun 2024 07:30:53 -0700 Subject: [PATCH] test(test_streaming.py): reduce vertex tests to avoid exhausting limits --- litellm/tests/test_streaming.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 7eafa8a6040..622b2efc84d 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -1035,7 +1035,7 @@ def test_completion_claude_stream_bad_key(): # test_completion_replicate_stream() -@pytest.mark.parametrize("provider", ["vertex_ai", "vertex_ai_beta"]) +@pytest.mark.parametrize("provider", ["vertex_ai"]) # "vertex_ai_beta" def test_vertex_ai_stream(provider): from litellm.tests.test_amazing_vertex_completion import load_vertex_ai_credentials