Lint fixes

microsoft · Oct 11, 2024 · 18862b9 · 18862b9
1 parent f816280
commit 18862b9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
+    return value.const_value