diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py index 45a8b1eed730..922ff15562ea 100644 --- a/examples/graphbolt/pyg/hetero/node_classification.py +++ b/examples/graphbolt/pyg/hetero/node_classification.py @@ -345,10 +345,10 @@ def parse_args(): "igb-het-small", "igb-het-medium", "igb-het-large", - "igb-het-MLPerf", + "igb-het-mlperf", "igb-het", ], - help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|MLPerf].", + help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|mlperf].", ) parser.add_argument( "--fanout", diff --git a/examples/graphbolt/rgcn/hetero_rgcn.py b/examples/graphbolt/rgcn/hetero_rgcn.py index 36fc1185c324..c3a6a36c3dbe 100644 --- a/examples/graphbolt/rgcn/hetero_rgcn.py +++ b/examples/graphbolt/rgcn/hetero_rgcn.py @@ -671,10 +671,10 @@ def main(args): "igb-het-medium", "igb-het-large", "igb-het", - "igb-het-MLPerf", + "igb-het-mlperf", ], help="Dataset name. Possible values: ogbn-mag, ogb-lsc-mag240m, " - "igb-het, and igb-het-[tiny|small|medium|large|MLPerf].", + "igb-het, and igb-het-[tiny|small|medium|large|mlperf].", ) parser.add_argument("--num_epochs", type=int, default=3) parser.add_argument("--num_workers", type=int, default=0) diff --git a/python/dgl/graphbolt/impl/ondisk_dataset.py b/python/dgl/graphbolt/impl/ondisk_dataset.py index 855e964f83fd..a8a3972fdb6d 100644 --- a/python/dgl/graphbolt/impl/ondisk_dataset.py +++ b/python/dgl/graphbolt/impl/ondisk_dataset.py @@ -1052,8 +1052,8 @@ class BuiltinDataset(OnDiskDataset): "igb-het-large-seeds", "igb-het", "igb-het-seeds", - "igb-het-MLPerf", - "igb-het-MLPerf-seeds", + "igb-het-mlperf", + "igb-het-mlperf-seeds", ] _all_datasets = _datasets + _large_datasets