(self, gpb)
| 772 | self._init_metadata() |
| 773 | |
| 774 | def _init(self, gpb): |
| 775 | self._client = get_kvstore() |
| 776 | assert ( |
| 777 | self._client is not None |
| 778 | ), "Distributed module is not initialized. Please call dgl.distributed.initialize." |
| 779 | self._g = _get_graph_from_shared_mem( |
| 780 | self.graph_name, self._use_graphbolt |
| 781 | ) |
| 782 | self._gpb = get_shared_mem_partition_book(self.graph_name) |
| 783 | if self._gpb is None: |
| 784 | self._gpb = gpb |
| 785 | self._client.map_shared_data(self._gpb) |
| 786 | |
| 787 | def _init_ndata_store(self): |
| 788 | """Initialize node data store.""" |
no test coverage detected