From c38146e180dc2da4e28f82f55947eefd61f9fcf8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 16 Apr 2025 20:13:31 -0700 Subject: [PATCH] test fix --- tests/local_testing/test_text_completion.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index 55d6e229a3..697c3fc710 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -4116,12 +4116,11 @@ async def test_async_text_completion_chat_model_stream(): @pytest.mark.asyncio async def test_completion_codestral_fim_api(model): try: - if model == "vertex_ai/codestral@2405": - from test_amazing_vertex_completion import ( - load_vertex_ai_credentials, - ) + from test_amazing_vertex_completion import ( + load_vertex_ai_credentials, + ) - load_vertex_ai_credentials() + load_vertex_ai_credentials() litellm.set_verbose = True import logging @@ -4157,17 +4156,16 @@ async def test_completion_codestral_fim_api(model): @pytest.mark.parametrize( "model", - ["vertex_ai/codestral@2405"], + ["vertex_ai/mistral-nemo@2407"], ) @pytest.mark.asyncio async def test_completion_codestral_fim_api_stream(model): try: - if model == "vertex_ai/codestral@2405": - from test_amazing_vertex_completion import ( - load_vertex_ai_credentials, - ) + from test_amazing_vertex_completion import ( + load_vertex_ai_credentials, + ) - load_vertex_ai_credentials() + load_vertex_ai_credentials() import logging from litellm._logging import verbose_logger