Merge pull request #270 from RaulPPelaez/simplify_data

Small fix for OSX
torchmd · Feb 1, 2024 · d921666 · d921666
2 parents 7686fd8 + 35f9ebf
commit d921666
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/data.py b/torchmdnet/data.py
@@ -39,7 +39,7 @@ def get(self, idx):
         return data
 
     def __getattr__(self, __name):
-        return getattr(self.__dict__["_dataset"], __name)
+        return getattr(self.__getattribute__("_dataset"), __name)
 
 
 class DataModule(LightningDataModule):