Fix lint issues

microsoft · Dec 23, 2024 · 766791d · 766791d
1 parent 9a4a58e
commit 766791d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/onnxscript/rewriter/_ir_utils.py b/onnxscript/rewriter/_ir_utils.py
@@ -95,7 +95,7 @@ def is_singleton_value(
         return expected == scalar
     # rtol must be specified for float comparison
     assert rtol is not None
-    return math.isclose(scalar, expected, rtol=rtol)
+    return math.isclose(scalar, expected, rel_tol=rtol)
 
 
 def has_rank(value: ir.Value | None, rank: int) -> bool:

diff --git a/onnxscript/rewriter/pattern.py b/onnxscript/rewriter/pattern.py
@@ -1178,6 +1178,8 @@ def _multi_match(self, candidate: Iterable[ir.Node], check_removable: bool) -> M
 
         Args:
             candidate: An iterable of nodes that will be matched against the pattern output nodes.
+            check_removable: If True, check that the matched nodes can be removed (that is, that
+                they are not used elsewhere in the graph).
         """
         match = self._match
         for pattern_node, node in zip(self.pattern.output_nodes, candidate):