Skip to content

Commit

Permalink
Update examples/graphbolt/pyg/hetero/node_classification.py
Browse files Browse the repository at this point in the history
Co-authored-by: Muhammed Fatih BALIN <[email protected]>
  • Loading branch information
BowenYao18 and mfbalin committed Sep 6, 2024
1 parent ebdb01f commit d50977d
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion examples/graphbolt/pyg/hetero/node_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,6 @@ def create_dataloader(
# }
node_feature_keys = {"paper": ["feat"], "author": ["feat"]}
if args.dataset == "ogb-lsc-mag240m":
node_feature_keys["author"] = ["feat"]
node_feature_keys["institution"] = ["feat"]
if "igb-het" in args.dataset:
node_feature_keys["institute"] = ["feat"]
Expand Down

0 comments on commit d50977d

Please sign in to comment.