biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI] Bump mypy version (#34950)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2026-03-04 20:55:11 +00:00
committed by GitHub
parent 7eca859110
commit 17dc9c7fc9
13 changed files with 90 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/kernels/core/test_rotary_embedding.py
View File

@@ -62,7 +62,7 @@ def test_rotary_embedding_opcheck(
)
key = torch.randn_like(query) if use_key else None
query = query[..., :head_size]
key = key[..., :head_size] if use_key else None
key = key[..., :head_size] if key is not None else None
rotary_embedding_opcheck(rot, positions, query, key)
@@ -73,5 +73,5 @@ def test_rotary_embedding_opcheck(
rot,
positions,
query.flatten(start_dim=-2),
key.flatten(start_dim=-2) if use_key else None,
key.flatten(start_dim=-2) if key is not None else None,
)