Apply lintrunner

Author: neuropilot-captain

examples/mediatek/aot_utils/oss_utils/utils.py

@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import os
-from typing import Dict, Optional
+from typing import Optional
 
 import torch
 from executorch import exir

examples/mediatek/model_export_scripts/dcgan.py

@@ -9,14 +9,12 @@
 if os.getcwd() not in sys.path:
     sys.path.append(os.getcwd())
 import argparse
-import json
 import os
 
 import dcgan_main
-import numpy as np
 
 import torch
-from aot_utils.oss_utils.utils import build_executorch_binary, make_output_dir
+from aot_utils.oss_utils.utils import build_executorch_binary
 from executorch.backends.mediatek import Precision
 
 

examples/mediatek/model_export_scripts/emformer_rnnt.py

@@ -9,12 +9,8 @@
 if os.getcwd() not in sys.path:
     sys.path.append(os.getcwd())
 import argparse
-import json
 import os
 
-import numpy as np
-
-import torch
 from aot_utils.oss_utils.utils import build_executorch_binary
 from executorch.backends.mediatek import Precision
 from executorch.examples.models.emformer_rnnt import (

examples/mediatek/model_export_scripts/mobilebert.py

@@ -9,13 +9,9 @@
 if os.getcwd() not in sys.path:
     sys.path.append(os.getcwd())
 import argparse
-import json
 import os
 
-import numpy as np
-
-import torch
-from aot_utils.oss_utils.utils import build_executorch_binary, make_output_dir
+from aot_utils.oss_utils.utils import build_executorch_binary
 from executorch.backends.mediatek import Precision
 from executorch.examples.models.mobilebert import MobileBertModelExample
 

examples/mediatek/model_export_scripts/vit_b_16.py

@@ -9,12 +9,9 @@
 if os.getcwd() not in sys.path:
     sys.path.append(os.getcwd())
 import argparse
-import json
-
-import numpy as np
 
 import torch
-from aot_utils.oss_utils.utils import build_executorch_binary, make_output_dir
+from aot_utils.oss_utils.utils import build_executorch_binary
 from executorch.backends.mediatek import Precision
 from executorch.examples.models.torchvision_vit import TorchVisionViTModel
 
@@ -52,8 +49,8 @@ def forward(self, input1):
     instance = NhwcWrappedModel()
 
     # if dropout.p = 0, change probability to 1e-6 to prevent -inf when quantize
-    for name, module in instance.named_modules():
-        if type(module) == torch.nn.Dropout:
+    for _name, module in instance.named_modules():
+        if isinstance(module, torch.nn.Dropout):
             if module.p == 0:
                 module.p = 1e-6
 

examples/mediatek/model_export_scripts/wav2letter.py

@@ -9,12 +9,8 @@
 if os.getcwd() not in sys.path:
     sys.path.append(os.getcwd())
 import argparse
-import json
 
-import numpy as np
-
-import torch
-from aot_utils.oss_utils.utils import build_executorch_binary, make_output_dir
+from aot_utils.oss_utils.utils import build_executorch_binary
 from executorch.backends.mediatek import Precision
 from executorch.examples.models.wav2letter import Wav2LetterModel