Skip to content

Commit d040af0

Browse files
committed
fix lint
Signed-off-by: whx-sjtu <xiaowang990929@gmail.com>
1 parent b41cc3e commit d040af0

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

vllm/model_executor/layers/deepseek_v4_attention.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -460,9 +460,7 @@ def attention_impl(
460460
# overlap with default's GEMM + cache write.
461461
if self.indexer is not None:
462462
aux_stream = (
463-
self.aux_stream_list[0]
464-
if self.aux_stream_list is not None
465-
else None
463+
self.aux_stream_list[0] if self.aux_stream_list is not None else None
466464
)
467465
indexer = self.indexer
468466
# Local ref so the closure keeps a non-None type for mypy.
@@ -492,9 +490,7 @@ def wq_b_kv_insert_and_compress() -> torch.Tensor:
492490
elif self.compressor is not None:
493491
# wq_b + kv_insert on default, compressor on aux.
494492
aux_stream = (
495-
self.aux_stream_list[0]
496-
if self.aux_stream_list is not None
497-
else None
493+
self.aux_stream_list[0] if self.aux_stream_list is not None else None
498494
)
499495
compressor = self.compressor
500496

0 commit comments

Comments
 (0)