"src/vscode:/vscode.git/clone" did not exist on "f28cb9e118558595404160f2e66749bb7cafb836"
Unverified Commit 83ada296 authored by Triston's avatar Triston Committed by GitHub
Browse files

[Doc] Fix the doc for the partition_graph function (#4708)



* Fix the doc for the partition_graph func

* Lint for a long line

* Fix a lint
Co-authored-by: default avatarTriston Cao <tristonc@nvidia.com>
parent b1309217
...@@ -569,8 +569,9 @@ def partition_graph(g, graph_name, num_parts, out_path, num_hops=1, part_method= ...@@ -569,8 +569,9 @@ def partition_graph(g, graph_name, num_parts, out_path, num_hops=1, part_method=
... out_path='output/', reshuffle=True, ... out_path='output/', reshuffle=True,
... balance_ntypes=g.ndata['train_mask'], ... balance_ntypes=g.ndata['train_mask'],
... balance_edges=True) ... balance_edges=True)
>>> g, node_feats, edge_feats, gpb, graph_name = dgl.distributed.load_partition( >>> (
... 'output/test.json', 0) ... g, node_feats, edge_feats, gpb, graph_name, ntypes_list, etypes_list,
... ) = dgl.distributed.load_partition('output/test.json', 0)
''' '''
def get_homogeneous(g, balance_ntypes): def get_homogeneous(g, balance_ntypes):
if g.is_homogeneous: if g.is_homogeneous:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment