Unverified Commit 1fe08607 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bugfix] Fixes #2844 (#2847)


Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 66182b28
...@@ -871,8 +871,8 @@ class CoraBinary(DGLBuiltinDataset): ...@@ -871,8 +871,8 @@ class CoraBinary(DGLBuiltinDataset):
self.graphs, labels = load_graphs(str(graph_path)) self.graphs, labels = load_graphs(str(graph_path))
self.labels = [] self.labels = []
for i in range(len(lables)): for i in range(len(labels)):
self.labels.append(labels['{}'.format(i)].asnumpy()) self.labels.append(F.asnumpy(labels['{}'.format(i)]))
# load pmpds under self.raw_path # load pmpds under self.raw_path
with open("{}/pmpds.pkl".format(self.raw_path), 'rb') as f: with open("{}/pmpds.pkl".format(self.raw_path), 'rb') as f:
self.pmpds = _pickle_load(f) self.pmpds = _pickle_load(f)
......
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