From ebdb01f132bbad20ee54f9cb70117d770b02603e Mon Sep 17 00:00:00 2001 From: Bowen Yao <112051015+BowenYao18@users.noreply.github.com> Date: Thu, 5 Sep 2024 23:00:03 -0700 Subject: [PATCH] Update examples/graphbolt/pyg/hetero/node_classification.py Co-authored-by: Muhammed Fatih BALIN --- examples/graphbolt/pyg/hetero/node_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py index bfad1ab0cf4d..d231c94b7cbc 100644 --- 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"]