From b29261771b5335e80353339e045db86ecff63638 Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Sun, 18 Dec 2011 11:31:18 -0500 Subject: [PATCH] Use AttributeError in _AttrDict --- cuburn/genome.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cuburn/genome.py b/cuburn/genome.py index 6b60985..9efc19c 100644 --- a/cuburn/genome.py +++ b/cuburn/genome.py @@ -111,7 +111,9 @@ def palette_encode(data, format='rgb8'): class _AttrDict(dict): def __getattr__(self, name): - return self[name] + if name in self: + return self[name] + raise AttributeError('%s not a dict key' % name) @classmethod def _wrap(cls, dct):