Skip to content

GitLab

Unverified Commit 0cf99be3 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by GitHub
Browse files

hotfix (#2163) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-56-220.ec2.internal>
parent 101dc0de
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/dgl/data/tu.py
View file @ 0cf99be3
...@@ -323,6 +323,8 @@ class TUDataset(DGLBuiltinDataset): ...@@ -323,6 +323,8 @@ class TUDataset(DGLBuiltinDataset):
delimiter=',').astype(int) delimiter=',').astype(int)
if 'label' in filename: if 'label' in filename:
data = F.tensor(self._idx_from_zero(data)) data = F.tensor(self._idx_from_zero(data))
else:
data = F.tensor(data)
getattr(g, field_name[0])[field_name[1]] = data getattr(g, field_name[0])[field_name[1]] = data
except IOError: except IOError:
pass pass
......
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