fix: add markers to int tests

langchain-ai · Feb 29, 2024 · e0b27af · e0b27af
1 parent 85a1b40
commit e0b27af
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/vertexai/tests/integration_tests/test_embeddings.py b/libs/vertexai/tests/integration_tests/test_embeddings.py
@@ -81,18 +81,21 @@ def test_warning(caplog: pytest.LogCaptureFixture) -> None:
     assert record.message == expected_message
 
 
+@pytest.mark.release
 def test_langchain_google_vertexai_image_embeddings(tmp_image) -> None:
     model = VertexAIEmbeddings(model_name="multimodalembedding")
     output = model.embed_image(tmp_image)
     assert len(output) == 1408
 
 
+@pytest.mark.release
 def test_langchain_google_vertexai_text_model() -> None:
     embeddings_model = VertexAIEmbeddings(model_name="textembedding-gecko@001")
     assert isinstance(embeddings_model.client, TextEmbeddingModel)
     assert embeddings_model.model_type == GoogleEmbeddingModelType.TEXT
 
 
+@pytest.mark.release
 def test_langchain_google_vertexai_multimodal_model() -> None:
     embeddings_model = VertexAIEmbeddings(model_name="multimodalembedding@001")
     assert isinstance(embeddings_model.client, MultiModalEmbeddingModel)