fix lint

langchain-ai · Feb 26, 2024 · 19e73c2 · 19e73c2
1 parent 925f524
commit 19e73c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py b/libs/vertexai/langchain_google_vertexai/vectorstores/vectorstores.py
@@ -202,11 +202,11 @@ def _get_default_embeddings(cls) -> Embeddings:
         )
 
         # TODO: Change to vertexai embbedingss
-        from langchain_community import (
-            embeddings,  # type: ignore[import-not-found, unused-ignore]
+        from langchain_community.embeddings import (
+            TensorflowHubEmbeddings,  # type: ignore[import-not-found, unused-ignore]
         )
 
-        return embeddings.TensorflowHubEmbeddings()
+        return TensorflowHubEmbeddings()
 
     def _generate_unique_ids(self, number: int) -> List[str]:
         """Generates a list of unique ids of length `number`