[Doc] Hotfix for GraphSAINT Doc (dmlc#3898)

* Update graphsaint.py * Update graphsaint.py
anirband · Mar 31, 2022 · 11e910b · 11e910b
1 parent 35e66f4
commit 11e910b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/dgl/dataloading/graphsaint.py b/python/dgl/dataloading/graphsaint.py
@@ -61,8 +61,7 @@ class SAINTSampler(Sampler):
     >>> num_iters = 1000
     >>> sampler = SAINTSampler(mode='node', budget=6000)
     >>> # Assume g.ndata['feat'] and g.ndata['label'] hold node features and labels
-    >>> dataloader = DataLoader(g, torch.arange(num_iters), sampler, num_workers=4,
-    ...                         prefetch_ndata=['feat', 'label'])
+    >>> dataloader = DataLoader(g, torch.arange(num_iters), sampler, num_workers=4)
     >>> for subg in dataloader:
     ...     train_on(subg)
     """