mypy

Author: xadupre

_unittests/ut_reference/test_onnxruntime_evaluator.py

@@ -110,7 +110,7 @@ def test_constant_bool(self):
         got = ref.run(None, {})[0]
         self.assertEqual(got.dtype, np.bool_)
         self.assertEqual(got, True)
-        ref = OnnxruntimeEvaluator(node)
+        ref = OnnxruntimeEvaluator(node, opsets=21)
         got = ref.run(None, {})[0]
         self.assertEqual(len(ref._cache), 1)
         values = list(ref._cache.values())
@@ -133,7 +133,7 @@ def test_constant_bool_array(self):
         got = ref.run(None, {})[0]
         self.assertEqual(got.dtype, np.bool_)
         self.assertEqual(got[0], True)
-        ref = OnnxruntimeEvaluator(node)
+        ref = OnnxruntimeEvaluator(node, opsets=21)
         got = ref.run(None, {})[0]
         self.assertEqual(len(ref._cache), 1)
         values = list(ref._cache.values())
@@ -163,7 +163,7 @@ def test_constant_bool_input(self):
         self.assertEqual(got.dtype, np.bool_)
         self.assertEqual(got[0], True)
 
-        ref = OnnxruntimeEvaluator(node)
+        ref = OnnxruntimeEvaluator(node, opsets=21)
         got = ref.run(None, feeds)[0]
         self.assertEqual(got.dtype, np.bool_)
         self.assertEqual(got[0], True)

onnx_diagnostic/reference/ort_evaluator.py

@@ -338,7 +338,7 @@ def _get_sess(
                 # We force the type to be a boolean.
                 ref = ExtendedReferenceEvaluator(node)
                 cst = ref.run(None, {})[0]
-                vinputs = []
+                vinputs: List[ValueInfoProto] = []
                 voutputs = [
                     oh.make_tensor_value_info(
                         node.output[0], dtype_to_tensor_dtype(cst.dtype), cst.shape