[Relay][vm] Small bug fix for DataTypeObject (#3604)

* small bug fix for DataTypeObject

* retrigger ci

diff --git a/python/tvm/relay/backend/vmobj.py b/python/tvm/relay/backend/vmobj.py
index d6b11604fb8bebf2b60ab03bc3061c31aa36c0af..4c92e9bf38a63a3a4378af0b65a0fdfb7d775601 100644 (file)
--- a/python/tvm/relay/backend/vmobj.py
+++ b/python/tvm/relay/backend/vmobj.py
@@ -85,7 +85,7 @@ class DatatypeObject(Object):
         return self.fields[idx]
 
     def __len__(self):
-        return self.num_fields
+        return len(self.fields)
 
     def __iter__(self):
         return iter(self.fields)

diff --git a/tests/python/relay/test_vm.py b/tests/python/relay/test_vm.py
index 706c8587596ac6792be32dc8826ee37b8ad405ea..fbb6ce55d0b97618208596f47bfcb8fc632de5df 100644 (file)
--- a/tests/python/relay/test_vm.py
+++ b/tests/python/relay/test_vm.py
@@ -213,6 +213,9 @@ def test_list_constructor():
     mod["main"] = f
 
     result = veval(mod)
+    assert len(result) == 2
+    assert len(result[1]) == 2
+
     obj = vmobj_to_list(result)
     tvm.testing.assert_allclose(obj, np.array([3,2,1]))