formatting

huggingface · Nov 1, 2024 · 5522333 · 5522333
1 parent d9c3fdc
commit 5522333
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
@@ -2471,7 +2471,18 @@ def test_speculative_sampling_target_distribution(self):
                 [
                     [-inf, 10.0, -inf, -inf, -inf, -inf, -inf, -inf, -inf, -inf],  # accepts 1
                     [-inf, -inf, -inf, -inf, 10.0, -inf, -inf, -inf, -inf, -inf],  # accepts 4
-                    [-inf, 2.0, -inf, 1.0, -inf, -inf, -inf, -0.01, 2.0, -inf],  # most likely to be 1 or 8, less likely to be 3, then 7, and should never be any other value
+                    [
+                        -inf,
+                        2.0,
+                        -inf,
+                        1.0,
+                        -inf,
+                        -inf,
+                        -inf,
+                        -0.01,
+                        2.0,
+                        -inf,
+                    ],  # most likely to be 1 or 8, less likely to be 3, then 7, and should never be any other value
                     [-inf, -inf, -inf, -inf, -inf, -inf, -inf, -inf, -inf, -inf],  # N/A
                 ]
             ]