style

huggingface · Sep 23, 2024 · 107d567 · 107d567
1 parent 86ffc13
commit 107d567
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
@@ -1908,7 +1908,9 @@ def test_new_cache_format(self, num_beams, do_sample):
 
             new_cache = new_results.past_key_values
             if cache_cls == DynamicSlidingWindowCache:
-                legacy_cache_converted = cache_cls.from_legacy_cache(config.sliding_window, legacy_results.past_key_values)
+                legacy_cache_converted = cache_cls.from_legacy_cache(
+                    config.sliding_window, legacy_results.past_key_values
+                )
             else:
                 legacy_cache_converted = cache_cls.from_legacy_cache(legacy_results.past_key_values)
             for layer_idx in range(len(new_cache)):