Function local_access

python/dgl/distributed/graph_services.py:1000–1026

(local_g, partition_book, local_nids)

        )

    def local_access(local_g, partition_book, local_nids):
        etype_offset = gpb.local_etype_offset
        # See NOTE 1
        if prob is not None and (not use_graphbolt):
            _prob = [
                (
                    g.edges[etype].data[prob].local_partition
                    if prob in g.edges[etype].data
                    else None
                )
                for etype in g.canonical_etypes
            ]
        else:
            _prob = prob
        return _sample_etype_neighbors(
            use_graphbolt,
            local_g,
            partition_book,
            local_nids,
            fanout,
            edge_dir=edge_dir,
            prob=_prob,
            exclude_edges=None,
            replace=replace,
            etype_offset=etype_offset,
            etype_sorted=etype_sorted,
        )

    frontier = _distributed_access(
        g, nodes, issue_remote_req, local_access, exclude_edges=exclude_edges