Skip to content

GitLab

Unverified Commit f1542b9d authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Fix (#1019)

parent d2afe65c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/dgl/convert.py
View file @ f1542b9d
...@@ -590,8 +590,8 @@ def to_homo(G): ...@@ -590,8 +590,8 @@ def to_homo(G):
srctype, _, dsttype = etype srctype, _, dsttype = etype
src, dst = G.all_edges(etype=etype, order='eid') src, dst = G.all_edges(etype=etype, order='eid')
num_edges = len(src) num_edges = len(src)
srcs.append(src + offset_per_ntype[G.get_ntype_id(srctype)]) srcs.append(src + int(offset_per_ntype[G.get_ntype_id(srctype)]))
dsts.append(dst + offset_per_ntype[G.get_ntype_id(dsttype)]) 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())) etype_ids.append(F.full_1d(num_edges, etype_id, F.int64, F.cpu()))
eids.append(F.arange(0, num_edges)) eids.append(F.arange(0, num_edges))
......
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