[CI] fix lint (#5216)

Fix CI lint error

- vLLM version: release/v0.13.0
- vLLM main:
ad32e3e19c

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
wangxiyuan
2025-12-20 17:03:25 +08:00
committed by GitHub
parent 74aa968a9f
commit bbde0f9743
2 changed files with 2 additions and 1 deletions

View File

@@ -26,7 +26,6 @@ ignore_missing_imports = True
[mypy-msprobe.*]
ignore_missing_imports = True
allow_untyped_imports = True
[mypy-xlite.*]
ignore_missing_imports = True

View File

@@ -71,6 +71,8 @@ class BudgetRefiner:
valid = group[group['cost'] <= slo_limit]
if not valid.empty:
max_row = valid.loc[valid['chunk_size'].idxmax()]
assert isinstance(ctx_len, int), "ctx_len must be an integer"
assert isinstance(d_num, int), "d_num must be an integer"
self.lookup[(ctx_len, d_num)] = int(max_row['chunk_size'])
self.context_keys.add(ctx_len)
self.dnum_keys.add(d_num)