Skip to content
Snippets Groups Projects
Unverified Commit ea07f740 authored by Tianqi Chen's avatar Tianqi Chen Committed by GitHub
Browse files

Update test_op_level1.py

parent 710af087
No related branches found
No related tags found
No related merge requests found
......@@ -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")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment