From 165c56e8d51e16acf5294fcd60876b0ea39fc81f Mon Sep 17 00:00:00 2001
From: Jameson Graef Rollins <jameson.rollins@ligo.org>
Date: Tue, 7 Jul 2020 13:30:05 -0700
Subject: [PATCH] handle plotting all ifos if inspiral_range package not
available.
---
gwinc/ifo/__main__.py | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/gwinc/ifo/__main__.py b/gwinc/ifo/__main__.py
index d2c55d74..cf9725aa 100644
--- a/gwinc/ifo/__main__.py
+++ b/gwinc/ifo/__main__.py
@@ -2,8 +2,6 @@ import argparse
import numpy as np
import matplotlib.pyplot as plt
-import inspiral_range
-
from . import IFOS, PLOT_STYLE
from .. import load_budget
@@ -38,11 +36,18 @@ def main():
for name, budget in budgets.items():
data = budget.calc()
- BNS_range = inspiral_range.range(freq, data)
- label = '{name:<{pad}} {bns:>6.0f} Mpc'.format(
+ try:
+ import inspiral_range
+ label_range = ' {:>6.0f} Mpc'.format(
+ inspiral_range.range(freq, data),
+ )
+ except ModuleNotFoundError:
+ label_range = ''
+
+ label = '{name:<{pad}}{range}'.format(
name=name,
pad=range_pad,
- bns=BNS_range,
+ range=label_range,
)
ax.loglog(freq, np.sqrt(data), label=label)
--
GitLab