diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index b908c7b5c..a4d5e8c0b 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -1088,6 +1088,7 @@ def test_completion_mistral_api(): pytest.fail(f"Error occurred: {e}") +@pytest.mark.skip(reason="backend api unavailable") @pytest.mark.asyncio async def test_completion_codestral_chat_api(): try: diff --git a/tests/local_testing/test_text_completion.py b/tests/local_testing/test_text_completion.py index 6059d60bc..c1e4d0e16 100644 --- a/tests/local_testing/test_text_completion.py +++ b/tests/local_testing/test_text_completion.py @@ -4109,9 +4109,7 @@ async def test_async_text_completion_chat_model_stream(): # asyncio.run(test_async_text_completion_chat_model_stream()) -@pytest.mark.parametrize( - "model", ["vertex_ai/codestral@2405", "text-completion-codestral/codestral-2405"] # -) +@pytest.mark.parametrize("model", ["vertex_ai/codestral@2405"]) # @pytest.mark.asyncio async def test_completion_codestral_fim_api(model): try: @@ -4154,7 +4152,7 @@ async def test_completion_codestral_fim_api(model): @pytest.mark.parametrize( "model", - ["vertex_ai/codestral@2405", "text-completion-codestral/codestral-2405"], + ["vertex_ai/codestral@2405"], ) @pytest.mark.asyncio async def test_completion_codestral_fim_api_stream(model):