[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:
1
mypy.ini
1
mypy.ini
@@ -26,7 +26,6 @@ ignore_missing_imports = True
|
||||
|
||||
[mypy-msprobe.*]
|
||||
ignore_missing_imports = True
|
||||
allow_untyped_imports = True
|
||||
|
||||
[mypy-xlite.*]
|
||||
ignore_missing_imports = True
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user