From 1294777526f78672563c5346590ed2aec3242f51 Mon Sep 17 00:00:00 2001
From: Sebastian Steinlechner <sebastian.steinlechner@ligo.org>
Date: Wed, 31 Mar 2021 23:58:47 +0200
Subject: [PATCH] add accumulate function to nb.Budget
---
gwinc/nb.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/gwinc/nb.py b/gwinc/nb.py
index 7671b7fd..55138883 100644
--- a/gwinc/nb.py
+++ b/gwinc/nb.py
@@ -332,6 +332,9 @@ class Budget(Noise):
references = []
"""List of reference noise classes, or (ref, cal) tuples"""
+ accumulate = quadsum
+ """Method used to calculate total noise trace"""
+
def __init__(self, freq=None, noises=None, **kwargs):
"""Initialize Budget object.
@@ -565,7 +568,7 @@ class Budget(Noise):
_precomp=_precomp,
)
budget.append(trace)
- total = quadsum([trace.psd for trace in budget if trace.name in self._budget_noises])
+ total = type(self).accumulate([trace.psd for trace in budget if trace.name in self._budget_noises])
return self._make_trace(
psd=total, budget=budget
)
--
GitLab