allow demo to pass in env

2015-03-13 22:27:36 -07:00 · 2015-03-13 22:27:36 -07:00 · 8cc650847a
commit 8cc650847a
parent fad4d69ee4
1 changed files with 1 additions and 1 deletions
--- a/tracker/rabit_demo.py
+++ b/tracker/rabit_demo.py
@ -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':
        cmd[0] = './' + cmd[0]
    cmd = ' '.join(cmd)
-    env = {}
+    env = os.environ.copy()
    for k, v in worker_env.items():
        env[k] = str(v)        
    env['rabit_task_id'] = str(taskid)