Unverified Commit e6f6c2eb authored by Kaiqiang Xu's avatar Kaiqiang Xu Committed by GitHub
Browse files

[Bugfix] Update deprecated method name in load_graph.py (#2769)



Method `dataset.num_labels` has been deprecated and replaced by `dataset.num_classes`.  
Updating the method name to avoid runtime warning.
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 0ff7127a
......@@ -9,7 +9,7 @@ def load_reddit():
g = data[0]
g.ndata['features'] = g.ndata['feat']
g.ndata['labels'] = g.ndata['label']
return g, data.num_labels
return g, data.num_classes
def load_ogb(name):
from ogb.nodeproppred import DglNodePropPredDataset
......
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