Fix tests

pinecone-io · Nov 14, 2024 · b2e7b54 · b2e7b54
1 parent d628e0e
commit b2e7b54
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pinecone/grpc/index_grpc.py b/pinecone/grpc/index_grpc.py
@@ -1,5 +1,5 @@
 import logging
-from typing import Optional, Dict, Union, List, Tuple, Any, TypedDict, Iterable, cast
+from typing import Optional, Dict, Union, List, Iterable, cast
 
 from google.protobuf import json_format
 
@@ -64,13 +64,15 @@ def __init__(
         config: Config,
         channel: Optional[Channel] = None,
         grpc_config: Optional[GRPCClientConfig] = None,
+        pool_threads: Optional[int] = None,
         _endpoint_override: Optional[str] = None,
     ):
         super().__init__(
             index_name=index_name,
             config=config,
             channel=channel,
             grpc_config=grpc_config,
+            pool_threads=pool_threads,
             _endpoint_override=_endpoint_override,
             use_asyncio=False,
         )

diff --git a/pinecone/grpc/index_grpc_asyncio.py b/pinecone/grpc/index_grpc_asyncio.py
@@ -54,13 +54,15 @@ def __init__(
         config: Config,
         channel: Optional[Channel] = None,
         grpc_config: Optional[GRPCClientConfig] = None,
+        pool_threads: Optional[int] = None,
         _endpoint_override: Optional[str] = None,
     ):
         super().__init__(
             index_name=index_name,
             config=config,
             channel=channel,
             grpc_config=grpc_config,
+            pool_threads=pool_threads,
             _endpoint_override=_endpoint_override,
             use_asyncio=True,
         )