From 38e862b592e18a05028f484cd7c4221b287bb7b5 Mon Sep 17 00:00:00 2001
From: Jameson Graef Rollins <jameson.rollins@ligo.org>
Date: Tue, 20 Oct 2020 07:40:18 -0700
Subject: [PATCH] only use budget noises when calculating budget total

This fixes a bug whereby references traces were being included in the sum
total.
---
 gwinc/nb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gwinc/nb.py b/gwinc/nb.py
index e6c406dd..19ce472e 100644
--- a/gwinc/nb.py
+++ b/gwinc/nb.py
@@ -543,7 +543,7 @@ class Budget(Noise):
                 _precomp=_precomp,
             )
             budget.append(trace)
-        total = quadsum([trace.psd for trace in budget])
+        total = quadsum([trace.psd for trace in budget if trace.name in self._budget_noises])
         return self._make_trace(
             psd=total, budget=budget
         )
-- 
GitLab