Revert "Simplify FloatCastDatasetWrapper"

This reverts commit 7726350.

torchmdnet/data.py

@@ -39,7 +39,12 @@ def get(self, idx):
         return data
 
     def __getattr__(self, name):
-        return getattr(self.dataset, name)
+        # Check if the attribute exists in the underlying dataset
+        if hasattr(self.dataset, name):
+            return getattr(self.dataset, name)
+        raise AttributeError(
+            f"'{type(self).__name__}' and its underlying dataset have no attribute '{name}'"
+        )
 
 
 class DataModule(LightningDataModule):