saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Raise output mismatch error in ort_test_dir_utils.py (#5364)

Browse source
This commit is contained in:
Chun-Wei Chen 2020-10-02 16:44:59 -07:00 committed by GitHub
parent f5e4c0ea04
commit 5bd7241839
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tools/python/ort_test_dir_utils.py
View file

@ -236,5 +236,7 @@ def run_test_dir(model_or_dir):
if not np.equal(expected, actual).all():
print('Mismatch for {}:\nExpected:{}\nGot:{}'.format(output_names[idx], expected, actual))
failed = True
print('FAILED' if failed else 'PASS')
if failed:
raise ValueError('FAILED due to output mismatch.')
else:
print('PASS')