lint

Author: shubhambhokare1

onnxscript/rewriter/ort_fusions/_whisper_tiny.py

@@ -55,7 +55,6 @@ def test_whisper_tiny(self):
             "/workspace/testing/whisper-opt/whisper-tiny-4.48/whisper-tiny_encoder_optimized.onnx",
         )
 
-        
         # Generate decoder model
         whisper_decoder_model = onnx.load(
             "/workspace/testing/whisper-opt/whisper-tiny-4.48/whisper-tiny_decoder.onnx"
@@ -85,7 +84,6 @@ def test_whisper_tiny(self):
             new_decoder_onnx_model,
             "/workspace/testing/whisper-opt/whisper-tiny-4.48/whisper-tiny_decoder_optimized.onnx",
         )
-        
 
         """
         test_with_ort = packaging.version.Version("1.20") <= ORT_VERSION

onnxscript/rewriter/ort_fusions/mha.py

@@ -5,7 +5,7 @@
 from typing import Sequence, Union
 
 import onnxscript.ir as ir
-from onnxscript.rewriter import _fusion_utils, _ir_utils, pattern
+from onnxscript.rewriter import _fusion_utils, pattern
 
 """
 The MultiHeadAttention pattern: generate an instance
@@ -285,7 +285,7 @@ def rewrite(
     ):
         num_heads = 64
         # TODO: (fix) Error caused by incorrect SDPA fusion for pre-scaling case
-        #num_heads = _ir_utils.get_dim(query_BSHDh, 2)
+        # num_heads = _ir_utils.get_dim(query_BSHDh, 2)
         if not isinstance(num_heads, int):
             return None