Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/view.py
View file @ 1a2b306f
...@@ -45,7 +45,7 @@ class NodeView(object): ...@@ -45,7 +45,7 @@ class NodeView(object):
def __call__(self): def __call__(self):
"""Return the nodes.""" """Return the nodes."""
return F.arange(0, len(self)) return F.arange(0, len(self), dtype=F.int64)
class NodeDataView(MutableMapping): class NodeDataView(MutableMapping):
__slot__ = ['_graph', '_nodes'] __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