Unit test fixes

pinecone-io · Nov 14, 2024 · 56a07d4 · 56a07d4
1 parent bbfd0eb
commit 56a07d4
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/pinecone/data/index.py b/pinecone/data/index.py
@@ -116,7 +116,7 @@ def __init__(
             api_client_klass=ApiClient,
             api_klass=VectorOperationsApi,
             config=self.config,
-            openapi_config=openapi_config,
+            openapi_config=self._openapi_config,
             pool_threads=pool_threads,
             api_version=API_VERSION,
         )

diff --git a/pinecone/data/query_results_aggregator.py b/pinecone/data/query_results_aggregator.py
@@ -1,8 +1,8 @@
 from typing import List, Tuple, Optional, Any, Dict
 import json
 import heapq
-from pinecone.core.openapi.data.models import Usage
-from pinecone.core.openapi.data.models import QueryResponse as OpenAPIQueryResponse
+from pinecone.core.openapi.db_data.models import Usage
+from pinecone.core.openapi.db_data.models import QueryResponse as OpenAPIQueryResponse
 
 from dataclasses import dataclass, asdict
 

diff --git a/poetry.lock b/poetry.lock