update (#2191)

1aa25cb6 · Chao Ma · GitHub · 76d66fd3 · 1aa25cb6
Unverified Commit 1aa25cb6 authored Sep 14, 2020 by Chao Ma Committed by GitHub Sep 14, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/graphsage/experimental/partition_graph.py examples/pytorch/graphsage/experimental/partition_graph.py +1 -1

No files found.
--- a/examples/pytorch/graphsage/experimental/partition_graph.py
+++ b/examples/pytorch/graphsage/experimental/partition_graph.py
@@ -42,7 +42,7 @@ if __name__ == '__main__':
        balance_ntypes = None

    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:
            sym_g.ndata[key] = g.ndata[key]
        g = sym_g