snsd0805/gpu-provider
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: update actions when the worker knows the master has leaved

Browse Source
This commit is contained in:
snsd0805 2024-05-31 01:40:18 +08:00
parent 4f035b6d53
commit dd67d88da0
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/communication.py
View File

@ -97,8 +97,13 @@ class ClusterCommunicationModule():
self.client_sock.send(f'[STOP_CHECK] {json.dumps(data)}'.encode()) self.client_sock.send(f'[STOP_CHECK] {json.dumps(data)}'.encode())
self.node_manager.docker_client.swarm.leave() self.node_manager.docker_client.swarm.leave()
self.node_manager.status = 'none' self.node_manager.status = 'none'
self.node_manager.actions = [
{'explanation': 'Add another node into our cluster', 'function': 'add_node'},
{'explanation': 'Exit', 'function': 'exit'},
]
print("You have leaved the cluster.") print("You have leaved the cluster.")
print("Please Enter to continued")
return False return False