Skip to content
This repository has been archived by the owner on Jan 13, 2024. It is now read-only.

Commit

Permalink
better messages
Browse files Browse the repository at this point in the history
  • Loading branch information
sdpython committed Jul 14, 2019
1 parent f828c19 commit 80184d4
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions mlprodict/onnxrt/validate.py
Original file line number Diff line number Diff line change
Expand Up @@ -607,7 +607,7 @@ def iterate_tqdm():
batch = 'max_rel_diff_batch' in obs and diff is not None
op1 = obs.get('domain_opset_', '')
op2 = obs.get('domain_opset_ai.onnx.ml', '')
op = '{}-{}'.format(op1, op2)
op = '{}/{}'.format(op1, op2)

if diff is not None:
if diff < 1e-5:
Expand Down Expand Up @@ -638,7 +638,7 @@ def iterate_tqdm():
log_exc = True
if len(excs) == 1:
name = excs[0][0]
if '_9ort_run_single_exc' in name or '_6ort_run_batch_exc' in name:
if '_9ort_run_single_exc' in name and '_6ort_run_batch_exc' not in name:
log_exc = False
if log_exc:
k, v = excs[0]
Expand Down Expand Up @@ -677,6 +677,9 @@ def summary_report(df):

def aggfunc(values):
if len(values) != 1:
values = [str(_).replace("\n", " ").replace('\r', '').strip(" ")
for _ in values]
values = [_ for _ in values if _]
vals = set(values)
if len(vals) != 1:
return " // ".join(map(str, values))
Expand Down

0 comments on commit 80184d4

Please sign in to comment.