From 6f66cd4a4dc8dcf61cb529ecdee6b4a8003e5aa6 Mon Sep 17 00:00:00 2001
From: rusty1s <matthias.fey@tu-dortmund.de>
Date: Mon, 23 Mar 2020 09:52:09 +0100
Subject: [PATCH] added collate_dgl to __init__

---
 ogb/graphproppred/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ogb/graphproppred/__init__.py b/ogb/graphproppred/__init__.py
index 8e66896c..db0b58f7 100644
--- 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