fix sin onnx ut

2023-03-07 11:18:30 +08:00 · 2023-03-07 11:18:30 +08:00 · 85590b1536
parent 8559dd7572
commit 85590b1536
1 changed files with 1 additions and 1 deletions
--- a/tests/st/ops/cpu/test_sin_op.py
+++ b/tests/st/ops/cpu/test_sin_op.py
@ -81,4 +81,4 @@ def test_sin_onnx():
    sess = ort.InferenceSession(onnx_model.SerializeToString())
    input_name = sess.get_inputs()[0].name
    result = sess.run([], {input_name: np_array})
-    assert np.allclose(list(ms_output)[0].asnumpy(), result[0])
+    assert np.allclose(ms_output.asnumpy(), result[0])