From f69395711361b523394164bb2ae5e66cd53dcec2 Mon Sep 17 00:00:00 2001
From: Christopher Wipf <wipf@ligo.mit.edu>
Date: Mon, 16 Nov 2020 15:15:54 -0800
Subject: [PATCH] BudgetTrace: missing attribute raises AttributeError not
KeyError
---
gwinc/trace.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/gwinc/trace.py b/gwinc/trace.py
index 175d5e46..5ff3aed7 100644
--- a/gwinc/trace.py
+++ b/gwinc/trace.py
@@ -53,7 +53,10 @@ class BudgetTrace:
return bdict
def __getattr__(self, name):
- return self._bdict[name]
+ try:
+ return self._bdict[name]
+ except KeyError:
+ raise AttributeError
def __getitem__(self, name):
"""get budget trace by name"""
--
GitLab