mypy

Author: xadupre

_unittests/ut_reference/test_torch_onnx_evaluator.py

@@ -1108,6 +1108,7 @@ def test_conv(self):
                 [oh.make_tensor_value_info("Y", TFLOAT, [None, None, None, None])],
             ),
             opset_imports=[oh.make_opsetid("", 18)],
+            ir_version=10,
         )
         sH, sW = 5, 6
         i = sH // 2
@@ -1130,6 +1131,7 @@ def test_nonzero(self):
                 [oh.make_tensor_value_info("Y", TINT64, [None, None])],
             ),
             opset_imports=[oh.make_opsetid("", 18)],
+            ir_version=10,
         )
 
         self._finalize_test(

onnx_diagnostic/reference/torch_ops/nn_ops.py

@@ -13,7 +13,7 @@ def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
         self.auto_pad = self.get_attribute_string(node, "auto_pad", "NOTSET")
         self.dilations = self.get_attribute_ints(node, "dilations", None)
         self.group = self.get_attribute_int(node, "group", 1)
-        self.kernel_shape = self.get_attribute_int(node, "kernel_shape", [])
+        self.kernel_shape = self.get_attribute_int(node, "kernel_shape") or []
         self.pads = self.get_attribute_ints(node, "pads", None)
         self.strides = self.get_attribute_ints(node, "strides", None)
 

onnx_diagnostic/reference/torch_ops/reduce_ops.py

@@ -30,7 +30,7 @@ def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
 class ReduceOpAxes(ReduceOp):
     def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
         super().__init__(node, version)
-        self.axes = self.get_attribute_ints(node, "axes", [])
+        self.axes = self.get_attribute_ints(node, "axes") or []
 
 
 class ReduceMax_18(ReduceOp):