commit 71fffd877b0126089dd2402a0ec449b3fc43013d Author: juga0 <juga@riseup.net> Date: Fri Aug 31 16:18:02 2018 +0000 Add arg to round to a num of digits when scaling as Torflow --- sbws/core/generate.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/sbws/core/generate.py b/sbws/core/generate.py index fe5459e..b937c5b 100644 --- a/sbws/core/generate.py +++ b/sbws/core/generate.py @@ -1,5 +1,5 @@ from sbws.globals import (fail_hard, SBWS_SCALE_CONSTANT, TORFLOW_SCALING, - SBWS_SCALING, TORFLOW_BW_MARGIN) + SBWS_SCALING, TORFLOW_BW_MARGIN, TORFLOW_ROUND_DIG) from sbws.lib.v3bwfile import V3BWFile from sbws.lib.resultdump import load_recent_results_in_datadir from argparse import ArgumentDefaultsHelpFormatter @@ -44,6 +44,10 @@ def gen_parser(sub): type=float, help="Cap maximum bw when scaling as Torflow. " "(Default: 0.05)") + p.add_argument('-r', '--torflow-round-digs', default=TORFLOW_ROUND_DIG, + type=int, + help="Number of most significant digits to round bw " + "when scaling as Torflow. (Default: 3)") def main(args, conf): @@ -77,7 +81,8 @@ def main(args, conf): state_fpath = conf.getpath('paths', 'state_fname') bw_file = V3BWFile.from_results(results, state_fpath, args.scale_constant, scaling_method, - torflow_cap=args.torflow_bw_margin) + torflow_cap=args.torflow_bw_margin, + torflow_round_digs=args.torflow_round_digs, output = args.output or \ conf.getpath('paths', 'v3bw_fname').format(now_fname()) bw_file.write(output)