Skip to content

Commit 21e935d

Browse files
committed
specdec_bench/utils.py: ruff format
Pre-commit ruff-format collapsed the dict comprehension onto a single line. Pure formatting — no behaviour change. Signed-off-by: Chenhan Yu <chenhany@nvidia.com>
1 parent e201fa3 commit 21e935d

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

  • examples/specdec_bench/specdec_bench

examples/specdec_bench/specdec_bench/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,7 @@ def get_tokenizer(path, trust_remote_code=False):
4545
kwargs = {"trust_remote_code": trust_remote_code}
4646
if isinstance(extra_special_tokens, list):
4747
kwargs["extra_special_tokens"] = {
48-
token.strip("<|>").replace("|", "_") + "_token": token
49-
for token in extra_special_tokens
48+
token.strip("<|>").replace("|", "_") + "_token": token for token in extra_special_tokens
5049
}
5150

5251
return AutoTokenizer.from_pretrained(path, **kwargs)

0 commit comments

Comments
 (0)