Add MACEOFF Dataset

torchmd · Jun 27, 2024 · f4308de · f4308de
1 parent e908988
commit f4308de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchmdnet/datasets/__init__.py b/torchmdnet/datasets/__init__.py
@@ -23,6 +23,7 @@
 from .qm9q import QM9q
 from .spice import SPICE
 from .genentech import GenentechTorsions
+from .maceoff import MACEOFF
 
 __all__ = [
     "Ace",
@@ -47,4 +48,5 @@
     "SPICE",
     "Tripeptides",
     "WaterBox",
+    "MACEOFF",
 ]