[GraphBolt] fix feat_size fetch issue (#6469)

a6daf822 · Rhett Ying · GitHub · 963f8559 · a6daf822
Unverified Commit a6daf822 authored Oct 19, 2023 by Rhett Ying Committed by GitHub Oct 19, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/sampling/graphbolt/rgcn/hetero_rgcn.py examples/sampling/graphbolt/rgcn/hetero_rgcn.py +1 -1

No files found.
--- a/examples/sampling/graphbolt/rgcn/hetero_rgcn.py
+++ b/examples/sampling/graphbolt/rgcn/hetero_rgcn.py
@@ -631,7 +631,7 @@ def main(args):
        args.dataset
    )

-    feat_size = features.size("node", None, "feat")[0]
+    feat_size = features.size("node", "paper", "feat")[0]

    # As `ogb-lsc-mag240m` is a large dataset, features of `author` and
    # `institution` are generated in advance and stored in the feature store.