Skip to content

GitLab

Unverified Commit d70a362d authored by Feiyang(Vance) Chen's avatar Feiyang(Vance) Chen Committed by GitHub
Browse files

Update shadow.py (#4034) 


Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
parent 14251504
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/dataloading/shadow.py
View file @ d70a362d
...@@ -43,7 +43,7 @@ class ShaDowKHopSampler(Sampler): ...@@ -43,7 +43,7 @@ class ShaDowKHopSampler(Sampler):
>>> dataloader = dgl.dataloading.DataLoader( >>> dataloader = dgl.dataloading.DataLoader(
... g, torch.arange(g.num_nodes()), sampler, ... g, torch.arange(g.num_nodes()), sampler,
... batch_size=5, shuffle=True, drop_last=False, num_workers=4) ... batch_size=5, shuffle=True, drop_last=False, num_workers=4)
>>> for input_nodes, output_nodes, (subgraph,) in dataloader: >>> for input_nodes, output_nodes, subgraph in dataloader:
... print(subgraph) ... print(subgraph)
... assert torch.equal(input_nodes, subgraph.ndata[dgl.NID]) ... assert torch.equal(input_nodes, subgraph.ndata[dgl.NID])
... assert torch.equal(input_nodes[:output_nodes.shape[0]], output_nodes) ... assert torch.equal(input_nodes[:output_nodes.shape[0]], output_nodes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment