Run lint

Author: gramalingam

onnxscript/optimizer/_legacy/_optimizer.py

@@ -8,16 +8,15 @@
 import onnx
 import onnx.shape_inference
 
-import onnxscript.optimizer
 from onnxscript import rewriter
-from onnxscript.optimizer._optimizer import _DEFAULT_REWRITE_RULES
-from onnxscript.optimizer._legacy.constant_folding import fold_constants
-from onnxscript.optimizer._remove_unused import remove_unused_nodes
-from onnxscript.optimizer._remove_unused_function import remove_unused_functions
 from onnxscript.optimizer._legacy._simple_function_folding import (
     inline_functions_with_unused_outputs,
     inline_simple_functions,
 )
+from onnxscript.optimizer._legacy.constant_folding import fold_constants
+from onnxscript.optimizer._optimizer import _DEFAULT_REWRITE_RULES
+from onnxscript.optimizer._remove_unused import remove_unused_nodes
+from onnxscript.optimizer._remove_unused_function import remove_unused_functions
 
 logger = logging.getLogger(__name__)
 

onnxscript/optimizer/_optimizer.py

@@ -5,9 +5,8 @@
 import logging
 
 from onnxscript import ir, rewriter
-from onnxscript.optimizer._remove_unused import remove_unused_nodes
 from onnxscript.optimizer import _constant_folding, _inliner
-
+from onnxscript.optimizer._remove_unused import remove_unused_nodes
 from onnxscript.rewriter import (
     broadcast_to_matmul,
     cast_constant_of_shape,

onnxscript/optimizer/_remove_unused.py

@@ -97,6 +97,7 @@ def _remove_unused_nodes(model: ir.Model) -> None:
 
     logger.info("Removed %s unused nodes", count)
 
+
 def remove_unused_nodes(model: ir.Model | onnx.ModelProto) -> None:
     """Removes unused nodes from a model."""
     if isinstance(model, ir.Model):