Really fix quality due to ruff release

2023-03-22 20:56:22 -04:00
parent ef28df0572
commit 80e3b36361
2 changed files with 8 additions and 6 deletions
--- a/tests/pipelines/test_pipelines_automatic_speech_recognition.py
+++ b/tests/pipelines/test_pipelines_automatic_speech_recognition.py
@@ -56,8 +56,10 @@ if is_torch_available():

@is_pipeline_test
 class AutomaticSpeechRecognitionPipelineTests(unittest.TestCase):
-    model_mapping = dict((list(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING.items()) if MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING else [])
-        + (MODEL_FOR_CTC_MAPPING.items() if MODEL_FOR_CTC_MAPPING else []))
+    model_mapping = dict(
+        (list(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING.items()) if MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING else [])
+        + (MODEL_FOR_CTC_MAPPING.items() if MODEL_FOR_CTC_MAPPING else [])
+    )

    def get_test_pipeline(self, model, tokenizer, processor):
        if tokenizer is None: