diff --git a/src/pylorax/base.py b/src/pylorax/base.py index ae6813c3..dad1a638 100644 --- a/src/pylorax/base.py +++ b/src/pylorax/base.py @@ -55,7 +55,10 @@ class DataHolder(dict): self[attr] = value def __getattr__(self, attr): - return self[attr] + if attr in self: + return self[attr] + else: + raise AttributeError def __setattr__(self, attr, value): self[attr] = value