fix test

apache · Oct 17, 2019 · eb938ad · eb938ad
1 parent 837b1e5
commit eb938ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/relay/test_adt.py b/tests/python/relay/test_adt.py
@@ -738,11 +738,11 @@ def run(dtype):
     run('int32')
 
 def vmobj_to_list(o):
-    if isinstance(o, tvm.relay.backend.vmobj.TensorObject):
+    if isinstance(o, tvm.relay.backend.vmobj.Tensor):
         return [o.asnumpy().tolist()]
     elif isinstance(o, tvm.relay.backend.interpreter.TensorValue):
         return [o.asnumpy()]
-    elif isinstance(o, tvm.relay.backend.vmobj.DatatypeObject):
+    elif isinstance(o, tvm.relay.backend.vmobj.Datatype):
         result = []
         for f in o:
             result.extend(vmobj_to_list(f))