Skip to content

Commit 7466753

Browse files
committed
fix test
1 parent 6b0a54a commit 7466753

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/python/relay/test_vm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,9 +59,9 @@ def veval(f, *args, ctx=tvm.cpu(), target="llvm"):
5959
return ret
6060

6161
def vmobj_to_list(o):
62-
if isinstance(o, tvm.relay.backend.vmobj.TensorObject):
62+
if isinstance(o, tvm.relay.backend.vm.Tensor):
6363
return [o.asnumpy().tolist()]
64-
elif isinstance(o, tvm.relay.backend.vmobj.DatatypeObject):
64+
elif isinstance(o, tvm.relay.backend.vm.Datatype):
6565
result = []
6666
for f in o:
6767
result.extend(vmobj_to_list(f))

0 commit comments

Comments
 (0)