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

Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
2026-06-03 19:34:33 +08:00 · 2024-08-30 11:56:52 -05:00
parent 647d8ea815
commit a71946539f
1 changed files with 3 additions and 3 deletions
--- 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")