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

Merge branch 'allow-override-of-budget-sum' into 'master'

add accumulate function to nb.Budget

See merge request gwinc/pygwinc!134
parents 20b38c29 12947775
No related branches found
No related tags found
1 merge request!134add accumulate function to nb.Budget
Pipeline #212980 passed
......@@ -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
)
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