From 300b9c4e99baed19707331303679548d16f7fa20 Mon Sep 17 00:00:00 2001
From: Sergio Vidiella Pinto <sergiovidiella@google.com>
Date: Thu, 29 Feb 2024 13:41:56 +0100
Subject: [PATCH] fix: ci lint and tests

---
 libs/vertexai/.gitignore                                 | 1 +
 libs/vertexai/tests/integration_tests/conftest.py        | 2 +-
 libs/vertexai/tests/integration_tests/test_embeddings.py | 4 ++--
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/libs/vertexai/.gitignore b/libs/vertexai/.gitignore
index bee8a64b7..7dcfd0e22 100644
--- 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
index feba4c073..b09033667 100644
--- 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
index 67aa09e7e..56fb03cf5 100644
--- 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,