From 57accb5441e1586cd42b19ee7236bef7be0b9548 Mon Sep 17 00:00:00 2001 From: shivesh Date: Sun, 6 May 2018 22:55:11 -0500 Subject: Sun May 6 22:55:11 CDT 2018 --- plot_sens.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'plot_sens.py') diff --git a/plot_sens.py b/plot_sens.py index 0c862eb..279d66a 100755 --- a/plot_sens.py +++ b/plot_sens.py @@ -331,27 +331,30 @@ def main(): if args.plot: print 'Plotting sensitivities' - basename = args.infile + '{0}/{1}'.format(args.likelihood, args.stat_method) + if args.likelihood in [Likelihood.GOLEMFIT, Likelihood.GF_FREQ]: + basename = args.infile + 'golemfit/{1}'.format(args.likelihood, args.stat_method) + elif args.likelihood is Likelihood.GAUSSIAN: + basename = args.infile + 'gaussian/{1}'.format(args.likelihood, args.stat_method) baseoutfile = basename[:5]+basename[5:].replace('data', 'plots') if args.run_method is SensitivityCateg.FULL: plot_utils.plot_sens_full( data = data, - outfile = basename + '/FULL', + outfile = baseoutfile + '/FULL', outformat = ['png'], args = args, ) elif args.run_method in fixed_angle_categ: plot_utils.plot_sens_fixed_angle( data = data, - outfile = basename + '/FIXED_ANGLE', + outfile = baseoutfile + '/FIXED_ANGLE', outformat = ['png'], args = args, ) elif args.run_method in corr_angles_categ: plot_utils.plot_sens_corr_angle( data = data, - outfile = basename + '/CORR_ANGLE', + outfile = baseoutfile + '/CORR_ANGLE', outformat = ['png'], args = args, ) -- cgit v1.2.3