Skip to content

Commit 2537811

Browse files
authored
Merge branch 'dmlc:master' into add-igbh-to-rgcn
2 parents de7dc90 + 0dc0d57 commit 2537811

25 files changed

+3439
-374
lines changed

examples/graphbolt/pyg/hetero/node_classification.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,11 @@ def create_dataloader(
6565
node_feature_keys["institute"] = ["feat"]
6666
node_feature_keys["fos"] = ["feat"]
6767
# Fetch node features for the sampled subgraph.
68-
datapipe = datapipe.fetch_feature(features, node_feature_keys)
68+
datapipe = datapipe.fetch_feature(
69+
features,
70+
node_feature_keys,
71+
overlap_fetch=args.overlap_feature_fetch,
72+
)
6973

7074
# Copy the data to the specified device.
7175
if need_copy:

0 commit comments

Comments
 (0)