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

update

Browse Source
This commit is contained in:
Newnius 2020-05-26 23:00:02 +08:00
parent 5804c0ce72
commit f43af24ecd
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
agent.py
View File

@ -211,10 +211,11 @@ class MyHandler(BaseHTTPRequestHandler):
exit_code, output = container.exec_run(['sh', '-c', script])
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
lock.acquire()
pending_tasks[token] = {'gpus': str(docker_gpus).split(','), 'gpu_mem': int(docker_gpu_mem)}
id2token[msg['id']] = token
lock.release()
if docker_wait == "1":
lock.acquire()
pending_tasks[token] = {'gpus': str(docker_gpus).split(','), 'gpu_mem': int(docker_gpu_mem)}
id2token[msg['id']] = token
lock.release()
if exit_code != 0:
msg["code"] = 1
msg["error"] = output.decode('utf-8').rstrip('\n')