Format

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

onnxscript/rewriter/__init__.py

@@ -22,22 +22,22 @@
 import onnx_ir.passes.common as common_passes
 
 from onnxscript import ir
-from onnxscript.rewriter.rules import basic_rules as _basic_rules
-from onnxscript.rewriter.rules import broadcast_to_matmul as _broadcast_to_matmul
-from onnxscript.rewriter.rules import cast_constant_of_shape as _cast_constant_of_shape
-from onnxscript.rewriter.rules import collapse_slices as _collapse_slices
-from onnxscript.rewriter.rules import fuse_pad_into_conv as _fuse_pad_into_conv
-from onnxscript.rewriter.rules import fuse_relus_clips as _fuse_relus_clips
-from onnxscript.rewriter.rules import no_op as _no_op
 from onnxscript.rewriter import pattern
-from onnxscript.rewriter.rules import redundant_scatter_nd as _redundant_scatter_nd
 from onnxscript.rewriter._basics import MatchContext, MatchingTracer, MatchResult, MatchStatus
 from onnxscript.rewriter._rewrite_rule import (
     RewriterContext,
     RewriteRule,
     RewriteRuleClassBase,
     RewriteRuleSet,
 )
+from onnxscript.rewriter.rules import basic_rules as _basic_rules
+from onnxscript.rewriter.rules import broadcast_to_matmul as _broadcast_to_matmul
+from onnxscript.rewriter.rules import cast_constant_of_shape as _cast_constant_of_shape
+from onnxscript.rewriter.rules import collapse_slices as _collapse_slices
+from onnxscript.rewriter.rules import fuse_pad_into_conv as _fuse_pad_into_conv
+from onnxscript.rewriter.rules import fuse_relus_clips as _fuse_relus_clips
+from onnxscript.rewriter.rules import no_op as _no_op
+from onnxscript.rewriter.rules import redundant_scatter_nd as _redundant_scatter_nd
 
 _ModelProtoOrIr = TypeVar("_ModelProtoOrIr", onnx.ModelProto, ir.Model)
 _DEFAULT_REWRITE_RULES: tuple[pattern.RewriteRule, ...] = (

onnxscript/rewriter/rules/fuse_relus_clips_test.py

@@ -15,12 +15,12 @@
     RewriteRule,
     testing,
 )
+from onnxscript.rewriter.rules import fuse_relus_clips
 from onnxscript.rewriter.rules.fuse_relus_clips import (
     fuse_successive_clip_relu_rule,
     fuse_successive_clip_rule,
     fuse_successive_relu_clip_rule,
 )
-from onnxscript.rewriter.rules import fuse_relus_clips
 
 
 class _FuseReluClipTestBase(unittest.TestCase):

onnxscript/rewriter/rules/matmul_add_to_gemm_test.py

@@ -10,8 +10,8 @@
 
 from onnxscript import ir
 from onnxscript.rewriter import MatchingTracer, MatchStatus, testing
-from onnxscript.rewriter.rules.matmul_add_to_gemm import matmul_add_to_gemm_rule
 from onnxscript.rewriter.rules import matmul_add_to_gemm
+from onnxscript.rewriter.rules.matmul_add_to_gemm import matmul_add_to_gemm_rule
 
 
 class _MatMulAddToGemmTestBase(unittest.TestCase):