Function issue_remote_req

python/dgl/distributed/graph_services.py:974–998

(node_ids)

        nodes = F.cat(homo_nids, 0)

    def issue_remote_req(node_ids):
        if prob is not None and (not use_graphbolt):
            # See NOTE 1
            _prob = [
                (
                    # NOTE (BarclayII)
                    # Currently DistGraph.edges[] does not accept canonical etype.
                    g.edges[etype].data[prob].kvstore_key
                    if prob in g.edges[etype].data
                    else ""
                )
                for etype in g.canonical_etypes
            ]
        else:
            _prob = prob
        return SamplingRequestEtype(
            node_ids,
            fanout,
            edge_dir=edge_dir,
            prob=_prob,
            exclude_edges=None,
            replace=replace,
            etype_sorted=etype_sorted,
            use_graphbolt=use_graphbolt,
        )

    def local_access(local_g, partition_book, local_nids):
        etype_offset = gpb.local_etype_offset