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

Add detailed assertion error message (#7232)

Browse source
This commit is contained in:
Chun-Wei Chen 2021-04-05 10:05:40 -07:00 committed by GitHub
parent 008065aab1
commit 3ee9b0ec4d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/python/ort_test_dir_utils.py
View file

@ -218,7 +218,7 @@ def run_test_dir(model_or_dir):
# e.g. ONNX test models 20190729\opset8\tf_mobilenet_v2_1.4_224
if len(output_names) == 1 and output_names[0] == '':
output_names = [o.name for o in sess.get_outputs()]
assert(len(output_names) == 1)
assert len(output_names) == 1, 'There should be single output_name.'
expected_outputs[output_names[0]] = expected_outputs['']
expected_outputs.pop('')