MCPcopy Index your code
hub / github.com/dmlc/dgl / EdgesRequest

Class EdgesRequest

python/dgl/distributed/graph_services.py:578–598  ·  view source on GitHub ↗

Edges Request

Source from the content-addressed store, hash-verified

576
577
578class EdgesRequest(Request):
579 """Edges Request"""
580
581 def __init__(self, edge_ids, order_id):
582 self.edge_ids = edge_ids
583 self.order_id = order_id
584
585 def __setstate__(self, state):
586 self.edge_ids, self.order_id = state
587
588 def __getstate__(self):
589 return self.edge_ids, self.order_id
590
591 def process_request(self, server_state):
592 local_g = server_state.graph
593 partition_book = server_state.partition_book
594 global_src, global_dst = _find_edges(
595 local_g, partition_book, self.edge_ids
596 )
597
598 return FindEdgeResponse(global_src, global_dst, self.order_id)
599
600
601class InDegreeRequest(Request):

Callers 1

issue_remote_reqFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected