From 65c9b807997474f94e535b20cc8a72dd4048d281 Mon Sep 17 00:00:00 2001
From: Bowen Yao <112051015+BowenYao18@users.noreply.github.com>
Date: Thu, 17 Oct 2024 21:19:34 -0500
Subject: [PATCH] Update node_classification.py

---
 examples/graphbolt/pyg/hetero/node_classification.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
index dcefb7358be9..f43a6f0609ba 100644
--- a/examples/graphbolt/pyg/hetero/node_classification.py
+++ b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -345,8 +345,10 @@ def parse_args():
             "igb-het-small",
             "igb-het-medium",
             "igb-het-large",
+            "igb-het-MLPerf",
+            "igb-het"
         ],
-        help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het-[tiny|small|medium|large].",
+        help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|MLPerf].",
     )
     parser.add_argument(
         "--fanout",