From 2f7f46c3f0e708f55aea30233f10f6821b55e237 Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Fri, 17 May 2024 17:57:51 +0200 Subject: [PATCH] fix gemma test category (#232) --- libs/vertexai/tests/integration_tests/test_gemma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/vertexai/tests/integration_tests/test_gemma.py b/libs/vertexai/tests/integration_tests/test_gemma.py index 9d3e96d8..1b77c582 100644 --- a/libs/vertexai/tests/integration_tests/test_gemma.py +++ b/libs/vertexai/tests/integration_tests/test_gemma.py @@ -36,7 +36,7 @@ def test_gemma_model_garden() -> None: assert llm._llm_type == "gemma_vertexai_model_garden" -@pytest.mark.release +@pytest.mark.extended def test_gemma_chat_model_garden() -> None: """In order to run this test, you should provide endpoint names.