"""
import inspect
# Still depends on the legacy NodeCollator...
[docs]class DistNodeDataLoader(DistDataLoader): """Sampled graph data loader over nodes for distributed graph storage. It wraps an iterable over a set of nodes, generating the list of message flow graphs (MFGs) as computation dependency of the said minibatch, on a distributed graph. All the arguments have the same meaning as the single-machine counterpart :class:dgl.dataloading.DataLoader except the first argument :attr:g which must be a :class:dgl.distributed.DistGraph. Parameters ---------- g : DistGraph The distributed graph. nids, graph_sampler, device, kwargs : See :class:dgl.dataloading.DataLoader. See also -------- dgl.dataloading.DataLoader """
[docs]class DistEdgeDataLoader(DistDataLoader): """Sampled graph data loader over edges for distributed graph storage. It wraps an iterable over a set of edges, generating the list of message flow graphs (MFGs) as computation dependency of the said minibatch for edge classification, edge regression, and link prediction, on a distributed graph. All the arguments have the same meaning as the single-machine counterpart :class:dgl.dataloading.EdgeDataLoader except the first argument :attr:g which must be a :class:dgl.distributed.DistGraph. Parameters ---------- g : DistGraph The distributed graph. eids, graph_sampler, device, kwargs : See :class:dgl.dataloading.EdgeDataLoader. See also -------- dgl.dataloading.EdgeDataLoader """