From 7572794addb75168dfb1db6f9863f5dda1616cbd Mon Sep 17 00:00:00 2001 From: tqchen Date: Mon, 12 Jan 2015 12:02:28 -0800 Subject: [PATCH] add stacklevel for rabit --- tracker/rabit_hadoop.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tracker/rabit_hadoop.py b/tracker/rabit_hadoop.py index 2883b1c47..d6adbb5cc 100755 --- a/tracker/rabit_hadoop.py +++ b/tracker/rabit_hadoop.py @@ -30,7 +30,7 @@ if hadoop_binary == None or hadoop_streaming_jar == None: warnings.warn('Warning: Cannot auto-detect path to hadoop and hadoop-streaming jar\n'\ '\tneed to set them via arguments -hs and -hb\n'\ '\tTo enable auto-detection, you can set enviroment variable HADOOP_HOME'\ - ', or modify rabit_hadoop.py line 14') + ', or modify rabit_hadoop.py line 14', stacklevel = 2) parser = argparse.ArgumentParser(description='Rabit script to submit rabit jobs using Hadoop Streaming.'\ 'This script support both Hadoop 1.0 and Yarn(MRv2), Yarn is recommended') @@ -122,7 +122,7 @@ def hadoop_streaming(nworker, worker_args, use_yarn): if args.nthread != -1: if kmap['nthread'] is None: warnings.warn('nthread can only be set in Yarn(Hadoop version greater than 2.0),'\ - 'it is recommended to use Yarn to submit rabit jobs') + 'it is recommended to use Yarn to submit rabit jobs', stacklevel = 2) else: cmd += ' -D%s=%d' % (kmap['nthread'], args.nthread) cmd += ' -D%s=%d' % (kmap['timeout'], args.timeout)