From 061b0e504e63fcd7777f9bcc0f8226b267146600 Mon Sep 17 00:00:00 2001 From: Ting-Jun Wang Date: Wed, 29 May 2024 18:43:42 +0800 Subject: [PATCH] fix: use host as IP --- src/communication.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/communication.py b/src/communication.py index 326b8b3..9be804f 100644 --- a/src/communication.py +++ b/src/communication.py @@ -11,7 +11,6 @@ class ClusterCommunicationModule(): self.client_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.host = host self.port = port - self.IP = socket.gethostbyname(socket.gethostname()) self.node_manager = node_manager self.worker_conns = [] @@ -66,7 +65,7 @@ class ClusterCommunicationModule(): def handle_command(self, data): command, args = data.split() if command == '[INFO]': - data = {'host': self.IP, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num} + data = {'host': self.host, 'GPU': self.node_manager.GPU, 'GPU_num': self.node_manager.GPU_num} self.client_sock.send(json.dumps(data).encode()) return True @@ -118,7 +117,6 @@ class ServiceExplorationModule(): self.host = host self.port = port - self.IP = socket.gethostbyname(socket.gethostname()) self.node_manager = node_manager @@ -131,7 +129,7 @@ class ServiceExplorationModule(): if self.node_manager.status == 'none': if data.decode() == '[EXPLORE]': - self.sock.sendto(self.IP.encode(), addr) + self.sock.sendto(self.host.encode(), addr) def explore(self): available_host = [] @@ -145,7 +143,7 @@ class ServiceExplorationModule(): while True: try: data, addr = client_sock.recvfrom(1024) - if addr[0] != self.IP: + if addr[0] != self.host: available_host.append(addr[0]) except: # if socket timeout