diff --git a/gwinc/gwinc.py b/gwinc/gwinc.py index 3327db156da222220b42eb37a0733b72324cd6ff..93b46d958a744bc85904f6e6f7ae62cdf9239152 100644 --- a/gwinc/gwinc.py +++ b/gwinc/gwinc.py @@ -32,12 +32,15 @@ def gwinc(freq, ifoin, source=None, plot=False, PRfixed=True): Budget, ifo_, freq_, plot_style = load_ifo('aLIGO') # add some precomputed info to the ifo struct - #this implicitly deepcopies and the return value is the copy + # this implicitly deepcopies and the return value is the copy ifo = precompIFO(freq, ifoin, PRfixed) traces = Budget(freq, ifo=ifo).calc_trace() - noises = {n:d[0] for n, d in traces.items()} + # construct matgwinc-compatible noises structure + noises = {} + for name, (data, style) in traces.items(): + noises[style.get('label', name)] = data noises['Freq'] = freq pbs = ifo.gwinc.pbs