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.