From 2db5e63bb617dea2e036d2d53f6ae805f2ffc2f4 Mon Sep 17 00:00:00 2001
From: Jameson Graef Rollins <jameson.rollins@ligo.org>
Date: Fri, 11 Dec 2020 10:28:20 -0800
Subject: [PATCH] cli: better budget name in plot title
---
gwinc/__main__.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/gwinc/__main__.py b/gwinc/__main__.py
index 98bdc1c8..0ee0c8f2 100644
--- a/gwinc/__main__.py
+++ b/gwinc/__main__.py
@@ -122,6 +122,7 @@ def main():
parser.exit(2, "IFO parameter specification not allowed when loading traces from file.\n")
from .io import load_hdf5
budget = None
+ name = args.IFO
trace = load_hdf5(args.IFO)
freq = trace.freq
ifo = trace.ifo
@@ -133,6 +134,7 @@ def main():
except IndexError:
parser.exit(2, "Improper frequency specification: {}\n".format(args.freq))
budget = load_budget(args.IFO, freq=freq)
+ name = budget.name
ifo = budget.ifo
plot_style = getattr(budget, 'plot_style', {})
trace = None
@@ -232,7 +234,7 @@ def main():
if args.title:
plot_style['title'] = args.title
else:
- plot_style['title'] = "GWINC Noise Budget: {}".format(args.IFO)
+ plot_style['title'] = "GWINC Noise Budget: {}".format(name)
if args.range:
logger.info("calculating inspiral ranges...")
--
GitLab