From 26b5fdac40aae50a4e9a38a5fffc6c80da6553cd Mon Sep 17 00:00:00 2001 From: chenshuaihua Date: Sun, 11 Jan 2015 23:54:31 +0800 Subject: [PATCH] yarn script --- tracker/rabit_yarn.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tracker/rabit_yarn.py b/tracker/rabit_yarn.py index 3ba8ddfc8..b5bd53677 100644 --- a/tracker/rabit_yarn.py +++ b/tracker/rabit_yarn.py @@ -71,10 +71,9 @@ parser.add_argument('command', nargs='+', help = 'command for rabit program') args = parser.parse_args() -if args.jobname is None: - args.jobname = ('Rabit(nworker=%d):' % args.nworker) + args.command[0].split('/')[-1]; - - +if args.jobname == 'auto': + args.jobname = ('Rabit[nworker=%d]:' % args.nworker) + args.command[0].split('/')[-1]; + def hadoop_streaming(nworker, slave_args): cmd = '%s jar %s -D mapreduce.job.maps=%d' % (args.hadoop_binary, args.hadoop_streaming_jar, nworker)