[Bugfix] Update deprecated method name in load_graph.py (dmlc#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: Minjie Wang <[email protected]>
skorani · Mar 22, 2021 · e6f6c2e · e6f6c2e
1 parent 0ff7127
commit e6f6c2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pytorch/graphsage/load_graph.py b/examples/pytorch/graphsage/load_graph.py
@@ -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