From eb938ad036155400ff3dd574ced9f0de54867f8a Mon Sep 17 00:00:00 2001 From: Wei Chen Date: Wed, 16 Oct 2019 19:38:12 -0700 Subject: [PATCH] fix test --- tests/python/relay/test_adt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/relay/test_adt.py b/tests/python/relay/test_adt.py index 48a233fb8e14..390d3cd9f3c4 100644 --- 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))