Update validate_difference.py

sdpython · Jul 12, 2019 · 2f392bb · 2f392bb
1 parent ec7e994
commit 2f392bb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mlprodict/onnxrt/validate_difference.py b/mlprodict/onnxrt/validate_difference.py
@@ -71,8 +71,9 @@ def measure_relative_difference(skl_pred, ort_pred, batch=True):
                 try:
                     ort_pred = numpy.array(ort_pred)
                 except ValueError as e:
-                    raise ValueError("Unable to interpret (type(skl_pred): {})\n{}\n-----\n{}".format(
-                        type(skl_pred), skl_pred, ort_pred)) from e
+                    raise ValueError(
+                        "Unable to interpret (batch={}, type(skl_pred): {})\n{}\n-----\n{}".format(
+                            batch, type(skl_pred), skl_pred, ort_pred)) from e
 
         if hasattr(skl_pred, 'todense'):
             skl_pred = skl_pred.todense()