Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow demo to pass in env

Browse Source
This commit is contained in:
tqchen 2015-03-13 22:27:36 -07:00
parent fad4d69ee4
commit 8cc650847a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tracker/rabit_demo.py
View File

@ -42,7 +42,7 @@ def exec_cmd(cmd, taskid, worker_env):
if cmd[0].find('/') == -1 and os.path.exists(cmd[0]) and os.name != 'nt': if cmd[0].find('/') == -1 and os.path.exists(cmd[0]) and os.name != 'nt':
cmd[0] = './' + cmd[0] cmd[0] = './' + cmd[0]
cmd = ' '.join(cmd) cmd = ' '.join(cmd)
env = {} env = os.environ.copy()
for k, v in worker_env.items(): for k, v in worker_env.items():
env[k] = str(v) env[k] = str(v)
env['rabit_task_id'] = str(taskid) env['rabit_task_id'] = str(taskid)