diff --git a/onnxscript/optimizer/_constant_folding.py b/onnxscript/optimizer/_constant_folding.py index 042b38247..818fd95e1 100644 --- a/onnxscript/optimizer/_constant_folding.py +++ b/onnxscript/optimizer/_constant_folding.py @@ -53,7 +53,7 @@ def is_constant_op(node: ir.Node) -> bool: # use ORT's implementation if we want to. -def _process_constant_node(node: ir.node) -> None: +def _process_constant_node(node: ir.Node) -> None: """Sets const_value of output value of a Constant op node.""" if node.op_type != "Constant" or node.domain not in {"", "ai.onnx"}: return @@ -660,7 +660,7 @@ def process_node(self, node: ir.Node): logger.debug( "Node [%s]: Replacing input %s with %s", node.name, - value.name, + value.name, # type: ignore[union-attr] sym_value.name, ) node.replace_input_with(i, sym_value) diff --git a/onnxscript/rewriter/_ir_utils.py b/onnxscript/rewriter/_ir_utils.py index 31cb8a30f..bd353f388 100644 --- a/onnxscript/rewriter/_ir_utils.py +++ b/onnxscript/rewriter/_ir_utils.py @@ -5,8 +5,9 @@ import onnxscript.ir as ir from onnxscript.optimizer import basic_constant_propagation + def get_const_value(value: ir.Value) -> ir.TensorProtocol | None: node = value.producer() if node is not None: basic_constant_propagation([node]) - return value.const_value \ No newline at end of file + return value.const_value