From 49b6e539b76c2c1f53f6e107496b32e4cdb592bf Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 21 Dec 2024 18:48:16 -0800 Subject: [PATCH] test fix --- tests/local_testing/test_amazing_vertex_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index d7a76e8542b..eadb9e093b0 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -343,7 +343,7 @@ async def test_aaavertex_ai_anthropic_async_streaming(): @pytest.mark.flaky(retries=3, delay=1) -def test_vertex_ai(): +def test_avertex_ai(): import random litellm.num_retries = 3 @@ -394,7 +394,7 @@ def test_vertex_ai(): @pytest.mark.flaky(retries=3, delay=1) -def test_vertex_ai_stream(): +def test_avertex_ai_stream(): load_vertex_ai_credentials() litellm.set_verbose = True litellm.vertex_project = "adroit-crow-413218"