YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-07 13:51:56 +00:00
Code Issues Releases Wiki Activity

update, robust

Browse Source
This commit is contained in:
Newnius 2020-07-06 10:54:31 +08:00
parent fe963127c5
commit 86aaeead30
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
agent.py
View File

@ -275,19 +275,19 @@ class MyHandler(BaseHTTPRequestHandler):
}) })
docker_image = form.getvalue('image') docker_image = form.getvalue('image')
docker_name = form.getvalue('name') docker_name = form.getvalue('name')
docker_cmd = form.getvalue('cmd') docker_cmd = form.getvalue('cmd', '')
docker_workspace = form.getvalue('workspace') docker_workspace = form.getvalue('workspace')
docker_gpus = form.getvalue('gpus') docker_gpus = form.getvalue('gpus', '')
docker_mem_limit = form.getvalue('mem_limit') docker_mem_limit = form.getvalue('mem_limit', 4096)
docker_cpu_limit = form.getvalue('cpu_limit') docker_cpu_limit = form.getvalue('cpu_limit', 4)
docker_network = form.getvalue('network') docker_network = form.getvalue('network')
docker_wait = form.getvalue('should_wait') docker_wait = form.getvalue('should_wait', '0')
docker_output = form.getvalue('output_dir') docker_output = form.getvalue('output_dir', '')
docker_hdfs_address = form.getvalue('hdfs_address') docker_hdfs_address = form.getvalue('hdfs_address', '')
docker_hdfs_dir = form.getvalue('hdfs_dir') docker_hdfs_dir = form.getvalue('hdfs_dir', '')
docker_gpu_mem = form.getvalue('gpu_mem') docker_gpu_mem = form.getvalue('gpu_mem', 8192)
dfs_src = form.getvalue('dfs_src') dfs_src = form.getvalue('dfs_src', '')
dfs_dst = form.getvalue('dfs_dst') dfs_dst = form.getvalue('dfs_dst', '')
token = generate_token(16) token = generate_token(16)
if len(dfs_src) > 0: if len(dfs_src) > 0: