1

Author: dkropachev

cassandra/connection.py

@@ -673,7 +673,7 @@ def __init__(self, start_port: int, end_port: int):
         self.start_port = start_port
         self.end_port = end_port
 
-    def generate(self, shard_id, total_shards):
+    def generate(self, shard_id: int, total_shards: int) -> Generator[int]:
         start = random.randrange(self.start_port, self.end_port)
         available_ports = itertools.chain(range(start, self.end_port), range(self.start_port, start))
         for port in available_ports: