Fix partition for 0.5.1 (dmlc#2187)

Co-authored-by: Ubuntu <[email protected]>
feixian15 · Sep 13, 2020 · 782527d · 782527d
1 parent 877101a
commit 782527d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pytorch/rgcn/experimental/partition_graph.py b/examples/pytorch/rgcn/experimental/partition_graph.py
@@ -50,7 +50,7 @@ def load_ogb(dataset, global_norm):
             if ntype == category:
                 category_id = i
 
-        g = dgl.to_homo(hg)
+        g = dgl.to_homogeneous(hg, edata=['norm'])
         if global_norm:
             u, v, eid = g.all_edges(form='all')
             _, inverse_index, count = th.unique(v, return_inverse=True, return_counts=True)