Unverified Commit 1a2b306f authored by Minjie Wang's avatar Minjie Wang Committed by GitHub
Browse files

[Bug] fix bug in nodes() that does not return int64 tensor (#116)

parent 68ec6247
......@@ -45,7 +45,7 @@ class NodeView(object):
def __call__(self):
"""Return the nodes."""
return F.arange(0, len(self))
return F.arange(0, len(self), dtype=F.int64)
class NodeDataView(MutableMapping):
__slot__ = ['_graph', '_nodes']
......
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