From 67b69b96b618e72e08088a26e591ee9b505dd4ad Mon Sep 17 00:00:00 2001 From: Jen Hamon Date: Fri, 15 Nov 2024 15:05:36 -0500 Subject: [PATCH] Fix tests broken by rename --- tests/unit/test_index.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_index.py b/tests/unit/test_index.py index c9037ff9..d3c8ff7c 100644 --- a/tests/unit/test_index.py +++ b/tests/unit/test_index.py @@ -2,7 +2,7 @@ import pytest import pinecone -from pinecone import Index +from pinecone.data import _Index from pinecone import UpsertRequest, Vector from pinecone import ( DescribeIndexStatsRequest, @@ -31,7 +31,7 @@ def setup_method(self): self.svv2 = [0.1, 0.2, 0.3] self.sv2 = {"indices": self.svi2, "values": self.svv2} - self.index = Index(api_key="asdf", host="https://test.pinecone.io") + self.index = _Index(api_key="asdf", host="https://test.pinecone.io") # region: upsert tests @@ -129,7 +129,7 @@ def test_upsert_parallelUpsert_callUpsertParallel(self, mocker): [Vector(id="vec1", values=self.vals1, metadata=self.md1)], [Vector(id="vec2", values=self.vals2, metadata=self.md2)], ] - with Index(api_key="asdf", host="https://test.pinecone.io", pool_threads=30) as index: + with _Index(api_key="asdf", host="https://test.pinecone.io", pool_threads=30) as index: mocker.patch.object(index._vector_api, "upsert_vectors", autospec=True) # Send requests in parallel