From 103444c58f66bd7a25407cb8fba6248d8cc976e5 Mon Sep 17 00:00:00 2001
From: "Quan (Andy) Gan" <coin2028@hotmail.com>
Date: Tue, 11 May 2021 14:06:18 +0800
Subject: [PATCH] Remove __len__ method from DGLGraph (#2902)

* Update heterograph.py

* remove unit tests

* replace tutorial
---
 python/dgl/heterograph.py        | 7 -------
 tests/compute/test_graph.py      | 2 --
 tutorials/models/1_gnn/4_rgcn.py | 2 +-
 3 files changed, 1 insertion(+), 10 deletions(-)

diff --git a/python/dgl/heterograph.py b/python/dgl/heterograph.py
index a1e88ad2ea2b..93a18fa9cca7 100644
--- a/python/dgl/heterograph.py
+++ b/python/dgl/heterograph.py
@@ -2447,13 +2447,6 @@ def num_edges(self, etype=None):
         else:
             return self._graph.number_of_edges(self.get_etype_id(etype))
 
-    def __len__(self):
-        """Deprecated: please directly call :func:`number_of_nodes`
-        """
-        dgl_warning('DGLGraph.__len__ is deprecated.'
-                    'Please directly call DGLGraph.number_of_nodes.')
-        return self.number_of_nodes()
-
     @property
     def is_multigraph(self):
         """Return whether the graph is a multigraph with parallel edges.
diff --git a/tests/compute/test_graph.py b/tests/compute/test_graph.py
index 8396420cefc2..5d25c7efcad7 100644
--- a/tests/compute/test_graph.py
+++ b/tests/compute/test_graph.py
@@ -59,7 +59,6 @@ def test_query():
     def _test_one(g):
         assert g.number_of_nodes() == 10
         assert g.number_of_edges() == 20
-        assert len(g) == 10
 
         for i in range(10):
             assert g.has_node(i)
@@ -131,7 +130,6 @@ def _test(g):
     def _test_csr_one(g):
         assert g.number_of_nodes() == 10
         assert g.number_of_edges() == 20
-        assert len(g) == 10
 
         for i in range(10):
             assert g.has_node(i)
diff --git a/tutorials/models/1_gnn/4_rgcn.py b/tutorials/models/1_gnn/4_rgcn.py
index 1f4d8e873632..18bb5b3795ff 100644
--- a/tutorials/models/1_gnn/4_rgcn.py
+++ b/tutorials/models/1_gnn/4_rgcn.py
@@ -313,7 +313,7 @@ def forward(self, g):
 g.edata.update({'rel_type': edge_type, 'norm': edge_norm})
 
 # create model
-model = Model(len(g),
+model = Model(g.num_nodes(),
               n_hidden,
               num_classes,
               num_rels,