Format array in test_transformer_lens.py (#275)

TransformerLensOrg · May 17, 2023 · 966ce1c · 966ce1c
1 parent c9236e4
commit 966ce1c
Showing 1 changed file with 5 additions and 59 deletions.
diff --git a/tests/acceptance/test_transformer_lens.py b/tests/acceptance/test_transformer_lens.py
@@ -66,65 +66,11 @@ def test_othello_gpt():
     sample_input = torch.tensor(
         [
             [
-                20,
-                19,
-                18,
-                10,
-                2,
-                1,
-                27,
-                3,
-                41,
-                42,
-                34,
-                12,
-                4,
-                40,
-                11,
-                29,
-                43,
-                13,
-                48,
-                56,
-                33,
-                39,
-                22,
-                44,
-                24,
-                5,
-                46,
-                6,
-                32,
-                36,
-                51,
-                58,
-                52,
-                60,
-                21,
-                53,
-                26,
-                31,
-                37,
-                9,
-                25,
-                38,
-                23,
-                50,
-                45,
-                17,
-                47,
-                28,
-                35,
-                30,
-                54,
-                16,
-                59,
-                49,
-                57,
-                14,
-                15,
-                55,
-                7,
+                # fmt: off
+                20, 19, 18, 10, 2, 1, 27, 3, 41, 42, 34, 12, 4, 40, 11, 29, 43, 13, 48, 56, 33,
+                39, 22, 44, 24, 5, 46, 6, 32, 36, 51, 58, 52, 60, 21, 53, 26, 31, 37, 9, 25, 38,
+                23, 50, 45, 17, 47, 28, 35, 30, 54, 16, 59, 49, 57, 14, 15, 55, 7,
+                # fmt: on
             ]
         ]
     )