mypy yapf

vllm-project · Jul 19, 2024 · 4b338d2 · 4b338d2
1 parent fa1f463
commit 4b338d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/vllm/spec_decode/interfaces.py b/vllm/spec_decode/interfaces.py
@@ -23,7 +23,7 @@ class SpeculativeProposals:
     proposal_lens: torch.Tensor
 
     # A flag to mark that there's no available proposals
-    no_proposals: bool
+    no_proposals: bool = False
 
     def __repr__(self):
         return (f"SpeculativeProposals("

diff --git a/vllm/spec_decode/top1_proposer.py b/vllm/spec_decode/top1_proposer.py
@@ -108,8 +108,7 @@ def get_spec_proposals(
             proposal_token_ids=proposal_tokens,
             proposal_probs=proposal_probs,
             proposal_lens=proposal_lens,
-            no_proposals=maybe_sampler_output is None
-        )
+            no_proposals=maybe_sampler_output is None)
 
         return proposals