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

* small bug fix for DataTypeObject * retrigger ci
wweic · Aug 9, 2019 · 31a8d76 · 31a8d76
1 parent b911eb4
commit 31a8d76
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/tvm/relay/backend/vmobj.py b/python/tvm/relay/backend/vmobj.py
@@ -85,7 +85,7 @@ def __getitem__(self, idx):
         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
@@ -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]))