From eb98a187b1bc4ee8e8c962d0e5d1a3d0ae77bc35 Mon Sep 17 00:00:00 2001 From: shivesh Date: Tue, 6 Nov 2018 17:49:17 -0600 Subject: plotting --- utils/plot.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'utils/plot.py') diff --git a/utils/plot.py b/utils/plot.py index fb58176..540ed25 100644 --- a/utils/plot.py +++ b/utils/plot.py @@ -34,7 +34,10 @@ from utils.enums import DataType, EnergyDependance from utils.enums import Likelihood, MixingScenario, ParamTag, StatCateg from utils.fr import angles_to_u, angles_to_fr -plt.style.use(os.environ['GOLEMSOURCEPATH']+'/GolemFit/scripts/paper/paper.mplstyle') +if os.environ.get('GOLEMSOURCEPATH') is not None: + plt.style.use(os.environ['GOLEMSOURCEPATH']+'/GolemFit/scripts/paper/paper.mplstyle') +elif os.path.isfile('./paper.mplstyle'): + plt.style.use('./paper.mplstyle') if 'submitter' in socket.gethostname(): rc('text', usetex=False) -- cgit v1.2.3