Skip to content

Commit 2b35ec4

Browse files
timofeev1995stainless-app[bot]
authored andcommitted
chore: fix lints (#299)
1 parent 16f64a5 commit 2b35ec4

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

examples/tokenize_data.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,10 @@ def pack_sequences(
7272
Sequence 3:
7373
['▁toys', '▁.', '</s>', '<s>', '▁but', '▁just', '▁one', '▁look']
7474
"""
75-
packed_sequences = []
76-
packed_position_ids = []
77-
buffer = []
78-
position_buffer = []
75+
packed_sequences: list[list[int]] = []
76+
packed_position_ids: list[list[int]] = []
77+
buffer: list[int] = []
78+
position_buffer: list[int] = []
7979

8080
for input_ids in batch["input_ids"]:
8181
# Truncate sequences that individually exceed max_seq_len (including EOS token).

0 commit comments

Comments
 (0)