Fix typing for address_remap (fixes redis#3234)

daveisfera · May 16, 2024 · 9343c0a · 9343c0a
1 parent 6751de2
commit 9343c0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/redis/cluster.py b/redis/cluster.py
@@ -506,7 +506,7 @@ def __init__(
         read_from_replicas: bool = False,
         dynamic_startup_nodes: bool = True,
         url: Optional[str] = None,
-        address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
+        address_remap: Optional[Callable[Tuple[str, int], Tuple[str, int]]] = None,
         **kwargs,
     ):
         """
@@ -1347,7 +1347,7 @@ def __init__(
         lock=None,
         dynamic_startup_nodes=True,
         connection_pool_class=ConnectionPool,
-        address_remap: Optional[Callable[[str, int], Tuple[str, int]]] = None,
+        address_remap: Optional[Callable[Tuple[str, int], Tuple[str, int]]] = None,
         **kwargs,
     ):
         self.nodes_cache = {}