Delegate missing attributes to original dataset object

msamogh · Jul 19, 2019 · 26382f0 · 26382f0
1 parent 69af86f
commit 26382f0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nonechucks/dataset.py b/nonechucks/dataset.py
@@ -92,3 +92,9 @@ def __getitem__(self, idx):
                 return sample
             idx += 1
         raise IndexError
+
+    def __getattr__(self, key):
+        """Delegates to original dataset object if an attribute is not
+        found in this class.
+        """
+        return getattr(self.dataset, key)