Update examples/graphbolt/node_classification.py

Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
dmlc · Aug 17, 2024 · 3d43ab8 · 3d43ab8
1 parent 418d06c
commit 3d43ab8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/graphbolt/node_classification.py b/examples/graphbolt/node_classification.py
@@ -371,9 +371,9 @@ def parse_args():
             "ogbn-arxiv",
             "ogbn-products",
             "ogbn-papers100M",
-            "igb-homogeneous-tiny",
-            "igb-homogeneous-small",
-            "igb-homogeneous-medium",
+            "igb-hom-tiny",
+            "igb-hom-small",
+            "igb-hom-medium",
         ],
         help="The dataset we can use for node classification example. Currently"
         " ogbn-products, ogbn-arxiv, ogbn-papers100M and"