mypy

xadupre · xadupre · commit 8a0546c4e32c · 2025-06-01T18:55:14.000+02:00
diff --git a/onnx_diagnostic/reference/torch_ops/nn_ops.py b/onnx_diagnostic/reference/torch_ops/nn_ops.py
@@ -13,7 +13,7 @@ def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
         self.axis = self.get_attribute_int(node, "axis", -1)
         self.epsilon = self.get_attribute_float(node, "epsilon", 1e-5)
         self.stash_type = onnx_dtype_to_torch_dtype(
-            self.get_attribute_int(node, "stash_type", onnx.TensorProto.FLOAT)
+            self.get_attribute_int(node, "stash_type", onnx.TensorProto.FLOAT)  # type: ignore[arg-type]
         )
         self.compute_std = len(node.output) > 1
 

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@ def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):`
`13`	`13`	`self.axis = self.get_attribute_int(node, "axis", -1)`
`14`	`14`	`self.epsilon = self.get_attribute_float(node, "epsilon", 1e-5)`
`15`	`15`	`self.stash_type = onnx_dtype_to_torch_dtype(`
`16`		`- self.get_attribute_int(node, "stash_type", onnx.TensorProto.FLOAT)`
	`16`	`+ self.get_attribute_int(node, "stash_type", onnx.TensorProto.FLOAT) # type: ignore[arg-type]`
`17`	`17`	`)`
`18`	`18`	`self.compute_std = len(node.output) > 1`
`19`	`19`