Skip to content

Commit b2e7b54

Browse files
committed
Fix tests
1 parent d628e0e commit b2e7b54

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

pinecone/grpc/index_grpc.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import logging
2-
from typing import Optional, Dict, Union, List, Tuple, Any, TypedDict, Iterable, cast
2+
from typing import Optional, Dict, Union, List, Iterable, cast
33

44
from google.protobuf import json_format
55

@@ -64,13 +64,15 @@ def __init__(
6464
config: Config,
6565
channel: Optional[Channel] = None,
6666
grpc_config: Optional[GRPCClientConfig] = None,
67+
pool_threads: Optional[int] = None,
6768
_endpoint_override: Optional[str] = None,
6869
):
6970
super().__init__(
7071
index_name=index_name,
7172
config=config,
7273
channel=channel,
7374
grpc_config=grpc_config,
75+
pool_threads=pool_threads,
7476
_endpoint_override=_endpoint_override,
7577
use_asyncio=False,
7678
)

pinecone/grpc/index_grpc_asyncio.py

+2
Original file line numberDiff line numberDiff line change
@@ -54,13 +54,15 @@ def __init__(
5454
config: Config,
5555
channel: Optional[Channel] = None,
5656
grpc_config: Optional[GRPCClientConfig] = None,
57+
pool_threads: Optional[int] = None,
5758
_endpoint_override: Optional[str] = None,
5859
):
5960
super().__init__(
6061
index_name=index_name,
6162
config=config,
6263
channel=channel,
6364
grpc_config=grpc_config,
65+
pool_threads=pool_threads,
6466
_endpoint_override=_endpoint_override,
6567
use_asyncio=True,
6668
)

0 commit comments

Comments
 (0)