diff --git a/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py b/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py index 36f4c7d3a210f..4be0445539270 100644 --- a/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py +++ b/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py @@ -1,6 +1,6 @@ from unittest.mock import Mock -from langchain_pinecone.vectorstores import Pinecone, PineconeVectorStore +from langchain_pinecone.vectorstores import PineconeVectorStore def test_initialization() -> None: @@ -9,7 +9,7 @@ def test_initialization() -> None: index = Mock() embedding = Mock() text_key = "xyz" - Pinecone(index, embedding, text_key) + PineconeVectorStore(index, embedding, text_key) def test_id_prefix() -> None: