add import check in __init__.py

dmlc · Sep 18, 2023 · be56669 · be56669
1 parent 464cbeb
commit be56669
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/dgl/data/__init__.py b/python/dgl/data/__init__.py
@@ -70,7 +70,12 @@
     RomanEmpireDataset,
     TolokersDataset,
 )
-from .lrgb import PeptidesStructuralDataset
+
+# RDKit is required for OGB-LSC PCQM4Mv2 and datasets derived from it.
+try:
+    from .lrgb import PeptidesStructuralDataset
+except ImportError:
+    pass
 from .pattern import PATTERNDataset
 from .wikics import WikiCSDataset
 from .yelp import YelpDataset