From 6abf527353733d99dbe3c13397f6fee622ace32f Mon Sep 17 00:00:00 2001
From: Evan Hall <evan.hall@ligo.org>
Date: Fri, 18 Oct 2019 15:51:16 -0400
Subject: [PATCH] Check if the trace is a sub-budget
---
gwinc/plot.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/gwinc/plot.py b/gwinc/plot.py
index 07ec4387..1b70bdce 100644
--- a/gwinc/plot.py
+++ b/gwinc/plot.py
@@ -1,5 +1,5 @@
from numpy import sqrt
-
+from collections import OrderedDict
def plot_noise(
freq,
@@ -24,6 +24,8 @@ def plot_noise(
ylim = kwargs.get('ylim')
for name, trace in traces.items():
+ if isinstance(trace, OrderedDict):
+ trace = trace['Total']
try:
data, style = trace
except:
--
GitLab