Update main_pyg.py

snap-stanford · Mar 23, 2020 · 3719ada · 3719ada
1 parent a1f2c04
commit 3719ada
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/examples/graphproppred/ppi/main_pyg.py b/examples/graphproppred/ppi/main_pyg.py
@@ -11,10 +11,7 @@
 
 ### importing OGB
 
-### for loading dataset
-from ogb.graphproppred.dataset_pyg import PygGraphPropPredDataset
-### for evaluation
-from ogb.graphproppred import Evaluator
+from ogb.graphproppred import PygGraphPropPredDataset, Evaluator
 
 multicls_criterion = torch.nn.CrossEntropyLoss()
 
@@ -149,4 +146,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()