From efb60be28963b55f5497a7ae20465d2354e9a8e0 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 26 May 2020 14:10:58 +0800 Subject: [PATCH] fix(mge/distributed): fix get_free_ports return value GitOrigin-RevId: 8a546855c43863d2cd6e96e7e9f7577811e1169d --- python_module/megengine/distributed/util.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/python_module/megengine/distributed/util.py b/python_module/megengine/distributed/util.py index 115ae326..ddeab73e 100644 --- a/python_module/megengine/distributed/util.py +++ b/python_module/megengine/distributed/util.py @@ -8,7 +8,7 @@ # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. import functools import socket -from typing import Callable, Optional +from typing import Callable, List, Optional import megengine._internal as mgb @@ -122,9 +122,8 @@ def synchronized(func: Callable): return wrapper -def get_free_ports(num: Optional[int] = 1) -> int: +def get_free_ports(num: int) -> List[int]: """Get one or more free ports. - Return an integer if num is 1, otherwise return a list of integers """ socks, ports = [], [] for i in range(num): @@ -134,4 +133,4 @@ def get_free_ports(num: Optional[int] = 1) -> int: ports.append(sock.getsockname()[1]) for sock in socks: sock.close() - return ports[0] if num == 1 else ports + return ports -- GitLab