Skip to content
Snippets Groups Projects
Commit 165c56e8 authored by Jameson Rollins's avatar Jameson Rollins
Browse files

handle plotting all ifos if inspiral_range package not available.

parent 7f814e08
No related branches found
No related tags found
No related merge requests found
...@@ -2,8 +2,6 @@ import argparse ...@@ -2,8 +2,6 @@ import argparse
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import inspiral_range
from . import IFOS, PLOT_STYLE from . import IFOS, PLOT_STYLE
from .. import load_budget from .. import load_budget
...@@ -38,11 +36,18 @@ def main(): ...@@ -38,11 +36,18 @@ def main():
for name, budget in budgets.items(): for name, budget in budgets.items():
data = budget.calc() data = budget.calc()
BNS_range = inspiral_range.range(freq, data) try:
label = '{name:<{pad}} {bns:>6.0f} Mpc'.format( 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, name=name,
pad=range_pad, pad=range_pad,
bns=BNS_range, range=label_range,
) )
ax.loglog(freq, np.sqrt(data), label=label) ax.loglog(freq, np.sqrt(data), label=label)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment