Merge pull request #8 from cblsjtu/master

correct a mistake
This commit is contained in:
Tianqi Chen 2015-01-14 10:07:17 -08:00
commit d05df9836b

View File

@ -103,7 +103,7 @@ def hadoop_streaming(nworker, worker_args, use_yarn):
args.command[i] = './' + args.command[i].split('/')[-1]
else:
args.command[i] = args.command[i].split('/')[-1]
if args.commands[0].endswith('.py'):
if args.command[0].endswith('.py'):
flst = [WRAPPER_PATH + '/rabit.py',
WRAPPER_PATH + '/librabit_wrapper.so',
WRAPPER_PATH + '/librabit_wrapper_mock.so']