mirror of
https://github.com/newnius/YAO-agent.git
synced 2025-12-16 06:06:44 +00:00
all in one
This commit is contained in:
1
main.py
1
main.py
@@ -148,6 +148,7 @@ class MyHandler(BaseHTTPRequestHandler):
|
|||||||
exit_code, output = container.exec_run('sh -c \'' + script + '\'')
|
exit_code, output = container.exec_run('sh -c \'' + script + '\'')
|
||||||
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
|
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
|
||||||
|
|
||||||
|
print(msg)
|
||||||
lock.acquire()
|
lock.acquire()
|
||||||
pending_tasks[msg['id']] = {'gpus': str(docker_gpus).split(',')}
|
pending_tasks[msg['id']] = {'gpus': str(docker_gpus).split(',')}
|
||||||
lock.release()
|
lock.release()
|
||||||
|
|||||||
Reference in New Issue
Block a user