Skip to content

Commit 866502b

Browse files
committed
Format: ruff format on convert_tiktoken.py and tokenizer.py
1 parent 41e7685 commit 866502b

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

scripts/convert_tiktoken.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def _decompose_token(mergeable_ranks: dict[bytes, int], token: bytes) -> list[by
4141
if min_idx is None:
4242
raise ValueError(f"Failed to find merge pair for token {token!r}")
4343

44-
parts = [*parts[:min_idx], parts[min_idx] + parts[min_idx + 1], *parts[min_idx + 2:]]
44+
parts = [*parts[:min_idx], parts[min_idx] + parts[min_idx + 1], *parts[min_idx + 2 :]]
4545

4646
return parts
4747

tinybpe/tokenizer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,9 @@ def __init__(
125125
self._special_tokens = special_tokens
126126
# Sort by length descending so that longer tokens match before
127127
# shorter prefixes (e.g. "<ab>" before "<a>").
128-
self._special_pattern = "(" + "|".join(
129-
re.escape(k) for k in sorted(special_tokens, key=len, reverse=True)
130-
) + ")"
128+
self._special_pattern = (
129+
"(" + "|".join(re.escape(k) for k in sorted(special_tokens, key=len, reverse=True)) + ")"
130+
)
131131

132132
if self._bytes_maps is None:
133133
_mapped = {k.encode("utf-8"): v for k, v in special_tokens.items()}

0 commit comments

Comments
 (0)