diff --git a/tests/python/unittest/test_ir_type.py b/tests/python/unittest/test_ir_type.py index a0e7d2b46ad6..1072efb11600 100644 --- a/tests/python/unittest/test_ir_type.py +++ b/tests/python/unittest/test_ir_type.py @@ -72,7 +72,7 @@ def test_func_type(): def test_tuple_type(): tp = tvm.ir.TypeVar('tp', tvm.ir.TypeKind.Type) - tf = tvm.ir.FuncType([], None, [], []) + tf = tvm.ir.FuncType([], tvm.ir.TupleType([]), [], []) tt = tvm.ir.TensorType(tvm.runtime.convert([1, 2, 3]), 'float32') fields = tvm.runtime.convert([tp, tf, tt])