[GraphBolt] Fix typo in node_classification example. (dmlc#7165)

ska278 · Feb 28, 2024 · ca35563 · ca35563
1 parent d1bea9e
commit ca35563
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/sampling/graphbolt/node_classification.py b/examples/sampling/graphbolt/node_classification.py
@@ -402,7 +402,7 @@ def main(args):
 
     # Load and preprocess dataset.
     print("Loading data...")
-    dataset = gb.BuiltinDataset("ogbn-products").load()
+    dataset = gb.BuiltinDataset(args.dataset).load()
 
     # Move the dataset to the selected storage.
     if args.storage_device == "pinned":