added collate_dgl to __init__

snap-stanford · Mar 23, 2020 · 6f66cd4 · 6f66cd4
1 parent 80be812
commit 6f66cd4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ogb/graphproppred/__init__.py b/ogb/graphproppred/__init__.py
@@ -8,5 +8,6 @@
 
 try:
     from .dataset_dgl import DglGraphPropPredDataset
+    from .dataset_dgl import collate_dgl
 except ImportError:
     pass