add temporal solution, run_yarn_prog.py

2015-03-13 21:13:19 -07:00
parent e5a9e31d13
commit d25de54008
4 changed files with 33 additions and 9 deletions
--- a/tracker/rabit_yarn.py
+++ b/tracker/rabit_yarn.py
@@ -96,7 +96,7 @@ def submit_yarn(nworker, worker_args, worker_env):
                if i == 0:
                    args.command[i] = './' + args.command[i].split('/')[-1]
                else:
-                    args.command[i] = args.command[i].split('/')[-1]
+                    args.command[i] = './' + args.command[i].split('/')[-1]
    if args.command[0].endswith('.py'):
        flst = [WRAPPER_PATH + '/rabit.py',
                WRAPPER_PATH + '/librabit_wrapper.so',