From fd1cca207f3c242fac69e8abc7adcd1e6fdb883f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 28 Aug 2024 21:12:39 -0700 Subject: [PATCH] test(test_amazing_vertex_completion.py): fix test --- litellm/tests/test_amazing_vertex_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 4c5c3dc8e0..02683de158 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -2112,7 +2112,7 @@ def test_get_token_url(): vertex_credentials=vertex_credentials, gemini_api_key="", custom_llm_provider="vertex_ai_beta", - should_use_vertex_v1beta1_features=should_use_v1beta1_features, + should_use_v1beta1_features=should_use_v1beta1_features, api_base=None, model="", stream=False, @@ -2132,7 +2132,7 @@ def test_get_token_url(): vertex_credentials=vertex_credentials, gemini_api_key="", custom_llm_provider="vertex_ai_beta", - should_use_vertex_v1beta1_features=should_use_v1beta1_features, + should_use_v1beta1_features=should_use_v1beta1_features, api_base=None, model="", stream=False,