fix: tests

tests/pipelines/test_pipelines_common.py

@@ -5,8 +5,8 @@
 import string
 import unittest
 from abc import abstractmethod
-from functools import lru_cache
 from unittest import skipIf
+from functools import lru_cache
 
 from transformers import (
     FEATURE_EXTRACTOR_MAPPING,
@@ -23,7 +23,6 @@
 from transformers.pipelines.base import _pad
 from transformers.testing_utils import is_pipeline_test, nested_simplify, require_tf, require_torch
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -153,8 +152,8 @@ def test(self):
                         if isinstance(model.config, (RobertaConfig, IBertConfig)):
                             tokenizer.model_max_length = model.config.max_position_embeddings - 2
                         elif (
-                            hasattr(model.config, "max_position_embeddings")
-                            and model.config.max_position_embeddings > 0
+                                hasattr(model.config, "max_position_embeddings")
+                                and model.config.max_position_embeddings > 0
                         ):
                             tokenizer.model_max_length = model.config.max_position_embeddings
                     # Rust Panic exception are NOT Exception subclass
@@ -591,4 +590,4 @@ def add(number, extra=0):
         dataset = PipelinePackIterator(dummy_dataset, add, {"extra": 2}, loader_batch_size=3)
 
         outputs = [item for item in dataset]
-        self.assertEqual(outputs, [[{"id": 2}, {"id": 3}, {"id": 4}, {"id": 5}]])
+        self.assertEqual(outputs, [[{"id": 2}, {"id": 3}, {"id": 4}, {"id": 5}]])