From ca09f4afec4b8a415c398b58dbb88fae8979176b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 23 Oct 2024 22:19:57 -0700 Subject: [PATCH] test: cleanup codestral tests - backend api unavailable --- tests/local_testing/test_completion.py | 1 + tests/local_testing/test_text_completion.py | 6 ++---- 2 files changed, 3 insertions(+), 4 deletions(-) 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):