Tensorflow examples fix (#5433)

examples/tensorflow/dgi/train.py

@@ -51,7 +51,7 @@ def main(args):
         test_mask = g.ndata["test_mask"]
         in_feats = features.shape[1]
         n_classes = data.num_labels
-        n_edges = data.graph.number_of_edges()
+        n_edges = g.number_of_edges()
 
         # add self loop
         if args.self_loop:

examples/tensorflow/gat/train.py

@@ -67,7 +67,7 @@ def main(args):
         test_mask = g.ndata["test_mask"]
         num_feats = features.shape[1]
         n_classes = data.num_labels
-        n_edges = data.graph.number_of_edges()
+        n_edges = g.number_of_edges()
         print(
             """----Data statistics------'
         #Edges %d

examples/tensorflow/gcn/train.py

@@ -44,7 +44,7 @@ def main(args):
         test_mask = g.ndata["test_mask"]
         in_feats = features.shape[1]
         n_classes = data.num_labels
-        n_edges = data.graph.number_of_edges()
+        n_edges = g.number_of_edges()
         print(
             """----Data statistics------'
         #Edges %d