Fix (#1019)

f1542b9d · Mufei Li · GitHub · d2afe65c · f1542b9d
Unverified Commit f1542b9d authored Nov 20, 2019 by Mufei Li Committed by GitHub Nov 20, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/dgl/convert.py python/dgl/convert.py +2 -2

No files found.
--- a/python/dgl/convert.py
+++ b/python/dgl/convert.py
@@ -590,8 +590,8 @@ def to_homo(G):
        srctype, _, dsttype = etype
        src, dst = G.all_edges(etype=etype, order='eid')
        num_edges = len(src)
-        srcs.append(src + offset_per_ntype[G.get_ntype_id(srctype)])
-        dsts.append(dst + offset_per_ntype[G.get_ntype_id(dsttype)])
+        srcs.append(src + int(offset_per_ntype[G.get_ntype_id(srctype)]))
+        dsts.append(dst + int(offset_per_ntype[G.get_ntype_id(dsttype)]))
        etype_ids.append(F.full_1d(num_edges, etype_id, F.int64, F.cpu()))
        eids.append(F.arange(0, num_edges))