diff --git a/examples/graphbolt/disk_based_feature/node_classification.py b/examples/graphbolt/disk_based_feature/node_classification.py index 144c5059a2..a1fd0f959e 100644 --- a/examples/graphbolt/disk_based_feature/node_classification.py +++ b/examples/graphbolt/disk_based_feature/node_classification.py @@ -340,6 +340,7 @@ def parse_args(): "igb-hom-small", "igb-hom-medium", "igb-hom-large", + "igb-hom", ], ) parser.add_argument("--root", type=str, default="datasets") diff --git a/examples/graphbolt/node_classification.py b/examples/graphbolt/node_classification.py index 810f41b679..4fff76e071 100644 --- a/examples/graphbolt/node_classification.py +++ b/examples/graphbolt/node_classification.py @@ -371,10 +371,11 @@ def parse_args(): "igb-hom-small", "igb-hom-medium", "igb-hom-large", + "igb-hom", ], help="The dataset we can use for node classification example. Currently" " ogbn-products, ogbn-arxiv, ogbn-papers100M and" - " igb-hom-[tiny|small|medium] datasets are supported.", + " igb-hom-[tiny|small|medium|large] and igb-hom datasets are supported.", ) parser.add_argument( "--mode", diff --git a/examples/graphbolt/pyg/labor/node_classification.py b/examples/graphbolt/pyg/labor/node_classification.py index e42aed10c3..f75fb439f0 100644 --- a/examples/graphbolt/pyg/labor/node_classification.py +++ b/examples/graphbolt/pyg/labor/node_classification.py @@ -367,6 +367,7 @@ def parse_args(): "igb-hom-small", "igb-hom-medium", "igb-hom-large", + "igb-hom", "reddit", "yelp", "flickr", diff --git a/examples/graphbolt/pyg/node_classification_advanced.py b/examples/graphbolt/pyg/node_classification_advanced.py index 743fc4d3cf..57de89011a 100644 --- a/examples/graphbolt/pyg/node_classification_advanced.py +++ b/examples/graphbolt/pyg/node_classification_advanced.py @@ -343,10 +343,11 @@ def parse_args(): "igb-hom-small", "igb-hom-medium", "igb-hom-large", + "igb-hom", ], help="The dataset we can use for node classification example. Currently" " ogbn-products, ogbn-arxiv, ogbn-papers100M and" - " igb-hom-[tiny|small|medium|large] datasets are supported.", + " igb-hom-[tiny|small|medium|large] and igb-hom datasets are supported.", ) parser.add_argument( "--fanout", diff --git a/python/dgl/graphbolt/impl/ondisk_dataset.py b/python/dgl/graphbolt/impl/ondisk_dataset.py index 3318491d28..6783bb296a 100644 --- a/python/dgl/graphbolt/impl/ondisk_dataset.py +++ b/python/dgl/graphbolt/impl/ondisk_dataset.py @@ -979,10 +979,11 @@ class BuiltinDataset(OnDiskDataset): .. note:: Reverse edges are added to the original graph. - **igb-hom-[tiny|small|medium|large]** - The igb-hom-[tiny|small|medium] dataset is a homogeneous citation network, - which is designed for developers to train and evaluate GNN models with - high fidelity. See more details in `igb-hom-[tiny|small|medium|large] + **igb-hom and igb-hom-[tiny|small|medium|large]** + The igb-hom-[tiny|small|medium|large] and igb-hom dataset is a homogeneous + citation network, which is designed for developers to train and evaluate + GNN models with high fidelity. See more details in + `igb-hom-[tiny|small|medium|large] `_. .. note:: @@ -1028,6 +1029,8 @@ class BuiltinDataset(OnDiskDataset): "igb-hom-medium-seeds", "igb-hom-large", "igb-hom-large-seeds", + "igb-hom", + "igb-hom-seeds", ] _all_datasets = _datasets + _large_datasets