diff --git a/tests/python/relay/test_op_level1.py b/tests/python/relay/test_op_level1.py index 61ac95ed8dc51e77b9eb86df716e3091ff7276b2..c7f8aa5ef63c7febd5ce973d60a48406a5021502 100644 --- a/tests/python/relay/test_op_level1.py +++ b/tests/python/relay/test_op_level1.py @@ -25,7 +25,7 @@ def test_softmax(): ib.ret(func) func = relay.ir_pass.infer_type(ib.env, func.to_func()) - ftype = func.checked_type() + ftype = func.checked_type assert ftype.ret_type == relay.ty.TensorType((n, d), "float32")