Skip to content

Commit fc116ea

Browse files
committed
Fix typing for address_remap (fixes redis#3234)
1 parent 6751de2 commit fc116ea

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

redis/asyncio/cluster.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ def __init__(
275275
ssl_min_version: Optional[ssl.TLSVersion] = None,
276276
ssl_ciphers: Optional[str] = None,
277277
protocol: Optional[int] = 2,
278-
address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
278+
address_remap: Optional[Callable[[Tuple[str, int]], Tuple[str, int]]] = None,
279279
cache_enabled: bool = False,
280280
client_cache: Optional[AbstractCache] = None,
281281
cache_max_size: int = 100,
@@ -1152,7 +1152,7 @@ def __init__(
11521152
startup_nodes: List["ClusterNode"],
11531153
require_full_coverage: bool,
11541154
connection_kwargs: Dict[str, Any],
1155-
address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
1155+
address_remap: Optional[Callable[[Tuple[str, int]], Tuple[str, int]]] = None,
11561156
) -> None:
11571157
self.startup_nodes = {node.name: node for node in startup_nodes}
11581158
self.require_full_coverage = require_full_coverage

redis/cluster.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -506,7 +506,7 @@ def __init__(
506506
read_from_replicas: bool = False,
507507
dynamic_startup_nodes: bool = True,
508508
url: Optional[str] = None,
509-
address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
509+
address_remap: Optional[Callable[[Tuple[str, int]], Tuple[str, int]]] = None,
510510
**kwargs,
511511
):
512512
"""
@@ -1347,7 +1347,7 @@ def __init__(
13471347
lock=None,
13481348
dynamic_startup_nodes=True,
13491349
connection_pool_class=ConnectionPool,
1350-
address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
1350+
address_remap: Optional[Callable[[Tuple[str, int]], Tuple[str, int]]] = None,
13511351
**kwargs,
13521352
):
13531353
self.nodes_cache = {}

0 commit comments

Comments
 (0)