Internal functions for loading DGLGraphs.
(filename, idx_list=None)
| 198 | |
| 199 | |
| 200 | def load_graph_v2(filename, idx_list=None): |
| 201 | """Internal functions for loading DGLGraphs.""" |
| 202 | if idx_list is None: |
| 203 | idx_list = [] |
| 204 | assert isinstance(idx_list, list) |
| 205 | heterograph_list = _CAPI_LoadGraphFiles_V2(filename, idx_list) |
| 206 | label_dict = load_labels_v2(filename) |
| 207 | return [gdata.get_graph() for gdata in heterograph_list], label_dict |
| 208 | |
| 209 | |
| 210 | def load_graph_v1(filename, idx_list=None): |
no test coverage detected