diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
index f8f699b6bf0a8c..7b7226ca3e62e9 100644
--- a/tests/generation/test_utils.py
+++ b/tests/generation/test_utils.py
@@ -60,7 +60,13 @@
         ImageGPTForCausalImageModeling,
         SpeechEncoderDecoderModel,
     )
-    from transformers.cache_utils import DynamicCache, DynamicSlidingWindowCache, EncoderDecoderCache, QuantoQuantizedCache, StaticCache
+    from transformers.cache_utils import (
+        DynamicCache,
+        DynamicSlidingWindowCache,
+        EncoderDecoderCache,
+        QuantoQuantizedCache,
+        StaticCache,
+    )
     from transformers.generation import (
         BeamSampleDecoderOnlyOutput,
         BeamSampleEncoderDecoderOutput,