YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-06 21:31:55 +00:00
Code Issues Releases Wiki Activity

all in one

Browse Source
This commit is contained in:
Newnius 2019-12-26 17:04:38 +08:00
parent e26711b49c
commit 93bdc81498
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -148,6 +148,7 @@ class MyHandler(BaseHTTPRequestHandler):
exit_code, output = container.exec_run('sh -c \'' + script + '\'')
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
print(msg)
lock.acquire()
pending_tasks[msg['id']] = {'gpus': str(docker_gpus).split(',')}
lock.release()