Skip to content
Snippets Groups Projects
Commit 12947775 authored by Sebastian Steinlechner's avatar Sebastian Steinlechner
Browse files

add accumulate function to nb.Budget

parent 20b38c29
No related branches found
No related tags found
1 merge request!134add accumulate function to nb.Budget
......@@ -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