fix style (#1715)

huggingface · Feb 23, 2024 · b5f4de3 · b5f4de3
1 parent 990c203
commit b5f4de3
Show file tree

Hide file tree

Showing 8 changed files with 32 additions and 19 deletions.
diff --git a/examples/onnxruntime/training/image-classification/run_image_classification.py b/examples/onnxruntime/training/image-classification/run_image_classification.py
@@ -23,6 +23,7 @@
 import evaluate
 import numpy as np
 import torch
+import transformers
 from datasets import load_dataset
 from PIL import Image
 from torchvision.transforms import (
@@ -35,8 +36,6 @@
     Resize,
     ToTensor,
 )
-
-import transformers
 from transformers import (
     MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING,
     AutoConfig,
@@ -51,6 +50,7 @@
 
 from optimum import ORTTrainer, ORTTrainingArguments
 
+
 """ Fine-tuning a 🤗 Transformers model for image classification"""
 
 logger = logging.getLogger(__name__)
@@ -447,4 +447,4 @@ def val_transforms(example_batch):
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/examples/onnxruntime/training/language-modeling/run_clm.py b/examples/onnxruntime/training/language-modeling/run_clm.py
@@ -598,9 +598,9 @@ def compute_metrics(eval_preds):
         # Data collator will default to DataCollatorWithPadding, so we change it.
         data_collator=default_data_collator,
         compute_metrics=compute_metrics if training_args.do_eval and not is_torch_tpu_available() else None,
-        preprocess_logits_for_metrics=preprocess_logits_for_metrics
-        if training_args.do_eval and not is_torch_tpu_available()
-        else None,
+        preprocess_logits_for_metrics=(
+            preprocess_logits_for_metrics if training_args.do_eval and not is_torch_tpu_available() else None
+        ),
     )
 
     # Training

diff --git a/examples/onnxruntime/training/language-modeling/run_mlm.py b/examples/onnxruntime/training/language-modeling/run_mlm.py
@@ -617,9 +617,9 @@ def compute_metrics(eval_preds):
         tokenizer=tokenizer,
         data_collator=data_collator,
         compute_metrics=compute_metrics if training_args.do_eval and not is_torch_tpu_available() else None,
-        preprocess_logits_for_metrics=preprocess_logits_for_metrics
-        if training_args.do_eval and not is_torch_tpu_available()
-        else None,
+        preprocess_logits_for_metrics=(
+            preprocess_logits_for_metrics if training_args.do_eval and not is_torch_tpu_available() else None
+        ),
     )
 
     # Training

diff --git a/tests/bettertransformer/test_audio.py b/tests/bettertransformer/test_audio.py
@@ -49,6 +49,7 @@ class BetterTransformersBarkTest(BetterTransformersTestMixin, unittest.TestCase)
     Since `Bark` is a text-to-speech model, it is preferrable
     to define its own testing class.
     """
+
     SUPPORTED_ARCH = ["bark"]
 
     FULL_GRID = {
@@ -185,6 +186,7 @@ class BetterTransformersAudioTest(BetterTransformersTestMixin, unittest.TestCase
     r"""
     Testing suite for Audio models - tests all the tests defined in `BetterTransformersTestMixin`
     """
+
     SUPPORTED_ARCH = ["wav2vec2", "hubert"]
 
     FULL_GRID = {

diff --git a/tests/bettertransformer/test_encoder.py b/tests/bettertransformer/test_encoder.py
@@ -37,6 +37,7 @@ class BetterTransformersEncoderTest(BetterTransformersTestMixin):
     - if the converted model produces the same logits as the original model.
     - if the converted model is faster than the original model.
     """
+
     SUPPORTED_ARCH = [
         "albert",
         "bert",

diff --git a/tests/bettertransformer/test_encoder_decoder.py b/tests/bettertransformer/test_encoder_decoder.py
@@ -35,6 +35,7 @@ class BetterTransformersEncoderDecoderTest(BetterTransformersTestMixin, unittest
     - if the converted model produces the same logits as the original model.
     - if the converted model is faster than the original model.
     """
+
     SUPPORTED_ARCH = [
         "bart",
         "blenderbot",

diff --git a/tests/bettertransformer/test_vision.py b/tests/bettertransformer/test_vision.py
@@ -27,6 +27,7 @@ class BetterTransformersVisionTest(BetterTransformersTestMixin, unittest.TestCas
     r"""
     Testing suite for Vision Models - tests all the tests defined in `BetterTransformersTestMixin`
     """
+
     SUPPORTED_ARCH = ["blip-2", "clip", "clip_text_model", "deit", "vilt", "vit", "vit_mae", "vit_msn", "yolos"]
 
     def prepare_inputs_for_class(self, model_id, model_type, batch_size=3, **preprocessor_kwargs):

diff --git a/tests/exporters/onnx/test_onnx_config_loss.py b/tests/exporters/onnx/test_onnx_config_loss.py
@@ -76,10 +76,12 @@ def test_onnx_config_with_loss(self):
                     ort_sess = onnxruntime.InferenceSession(
                         onnx_model_path.as_posix(),
                         providers=[
-                            "CUDAExecutionProvider"
-                            if torch.cuda.is_available()
-                            and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
-                            else "CPUExecutionProvider"
+                            (
+                                "CUDAExecutionProvider"
+                                if torch.cuda.is_available()
+                                and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
+                                else "CPUExecutionProvider"
+                            )
                         ],
                     )
                     framework = "pt" if isinstance(model, PreTrainedModel) else "tf"
@@ -145,9 +147,12 @@ def test_onnx_decoder_model_with_config_with_loss(self):
             ort_sess = onnxruntime.InferenceSession(
                 onnx_model_path.as_posix(),
                 providers=[
-                    "CUDAExecutionProvider"
-                    if torch.cuda.is_available() and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
-                    else "CPUExecutionProvider"
+                    (
+                        "CUDAExecutionProvider"
+                        if torch.cuda.is_available()
+                        and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
+                        else "CPUExecutionProvider"
+                    )
                 ],
             )
 
@@ -206,9 +211,12 @@ def test_onnx_seq2seq_model_with_config_with_loss(self):
             ort_sess = onnxruntime.InferenceSession(
                 onnx_model_path.as_posix(),
                 providers=[
-                    "CUDAExecutionProvider"
-                    if torch.cuda.is_available() and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
-                    else "CPUExecutionProvider"
+                    (
+                        "CUDAExecutionProvider"
+                        if torch.cuda.is_available()
+                        and "CUDAExecutionProvider" in onnxruntime.get_available_providers()
+                        else "CPUExecutionProvider"
+                    )
                 ],
             )
             batch = 3
-Original file line number
+Diff line change
@@ Expand Up @@
         r"""
         Testing suite for Vision Models - tests all the tests defined in `BetterTransformersTestMixin`
         """
         SUPPORTED_ARCH = ["blip-2", "clip", "clip_text_model", "deit", "vilt", "vit", "vit_mae", "vit_msn", "yolos"]
         def prepare_inputs_for_class(self, model_id, model_type, batch_size=3, **preprocessor_kwargs):
@@ Expand Down @@