fix: ci lint and tests

langchain-ai · Feb 29, 2024 · 300b9c4 · 300b9c4
1 parent 90796c4
commit 300b9c4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/libs/vertexai/.gitignore b/libs/vertexai/.gitignore
@@ -1 +1,2 @@
 __pycache__
+.mypy_cache_test
diff --git a/libs/vertexai/tests/integration_tests/conftest.py b/libs/vertexai/tests/integration_tests/conftest.py
@@ -2,7 +2,7 @@
 
 import pytest
 from _pytest.tmpdir import TempPathFactory
-from vertexai.vision_models import Image
+from vertexai.vision_models import Image  # type: ignore
 
 
 @pytest.fixture

diff --git a/libs/vertexai/tests/integration_tests/test_embeddings.py b/libs/vertexai/tests/integration_tests/test_embeddings.py
@@ -4,8 +4,8 @@
 `gcloud auth login` first).
 """
 import pytest
-from vertexai.language_models import TextEmbeddingModel
-from vertexai.vision_models import MultiModalEmbeddingModel
+from vertexai.language_models import TextEmbeddingModel  # type: ignore
+from vertexai.vision_models import MultiModalEmbeddingModel  # type: ignore
 
 from langchain_google_vertexai.embeddings import (
     GoogleEmbeddingModelType,