[Graphbolt] Add igb-hom-[tiny|small|medium] to disk_based_feature (#7759

) Co-authored-by: Muhammed Fatih BALIN <[email protected]>
dmlc · Aug 30, 2024 · a719465 · a719465
1 parent 647d8ea
commit a719465
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/graphbolt/disk_based_feature/node_classification.py b/examples/graphbolt/disk_based_feature/node_classification.py
@@ -336,9 +336,9 @@ def parse_args():
             "ogbn-arxiv",
             "ogbn-products",
             "ogbn-papers100M",
-            "reddit",
-            "yelp",
-            "flickr",
+            "igb-hom-tiny",
+            "igb-hom-small",
+            "igb-hom-medium",
         ],
     )
     parser.add_argument("--root", type=str, default="datasets")