Skip to content

GitLab

Unverified Commit 1aa25cb6 authored by Chao Ma's avatar Chao Ma Committed by GitHub
Browse files

update (#2191)

parent 76d66fd3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/graphsage/experimental/partition_graph.py
View file @ 1aa25cb6
...@@ -42,7 +42,7 @@ if __name__ == '__main__': ...@@ -42,7 +42,7 @@ if __name__ == '__main__':
balance_ntypes = None balance_ntypes = None
if args.undirected: if args.undirected:
sym_g = dgl.to_bidirected_stale(g, readonly=True) sym_g = dgl.to_bidirected(g, readonly=True)
for key in g.ndata: for key in g.ndata:
sym_g.ndata[key] = g.ndata[key] sym_g.ndata[key] = g.ndata[key]
g = sym_g g = sym_g
......
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