Skip to content

Commit eb938ad

Browse files
committed
fix test
1 parent 837b1e5 commit eb938ad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/python/relay/test_adt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -738,11 +738,11 @@ def run(dtype):
738738
run('int32')
739739

740740
def vmobj_to_list(o):
741-
if isinstance(o, tvm.relay.backend.vmobj.TensorObject):
741+
if isinstance(o, tvm.relay.backend.vmobj.Tensor):
742742
return [o.asnumpy().tolist()]
743743
elif isinstance(o, tvm.relay.backend.interpreter.TensorValue):
744744
return [o.asnumpy()]
745-
elif isinstance(o, tvm.relay.backend.vmobj.DatatypeObject):
745+
elif isinstance(o, tvm.relay.backend.vmobj.Datatype):
746746
result = []
747747
for f in o:
748748
result.extend(vmobj_to_list(f))

0 commit comments

Comments
 (0)