Update examples/graphbolt/pyg/hetero/node_classification.py

Co-authored-by: Muhammed Fatih BALIN <[email protected]>
dmlc · Sep 6, 2024 · ebdb01f · ebdb01f
1 parent aaf1da1
commit ebdb01f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -64,7 +64,7 @@ def create_dataloader(
     #         "author": ["feat"],
     #         "institution": ["feat"],
     #     }
-    node_feature_keys = {"paper": ["feat"]}
+    node_feature_keys = {"paper": ["feat"], "author": ["feat"]}
     if args.dataset == "ogb-lsc-mag240m":
         node_feature_keys["author"] = ["feat"]
         node_feature_keys["institution"] = ["feat"]