Skip to content

Commit

Permalink
Update node_classification.py
Browse files Browse the repository at this point in the history
  • Loading branch information
BowenYao18 authored Oct 18, 2024
1 parent 5dfd3fd commit 65c9b80
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion examples/graphbolt/pyg/hetero/node_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -345,8 +345,10 @@ def parse_args():
"igb-het-small",
"igb-het-medium",
"igb-het-large",
"igb-het-MLPerf",

Check warning on line 348 in examples/graphbolt/pyg/hetero/node_classification.py

View workflow job for this annotation

GitHub Actions / lintrunner

UFMT format

Run `lintrunner -a` to apply this patch.
"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",
Expand Down

0 comments on commit 65c9b80

Please sign in to comment.