Skip to content

Commit 71435ab

Browse files
committed
fix: check against None...
Signed-off-by: Travis Johnson <[email protected]>
1 parent b48fb5a commit 71435ab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_spyre/worker/spyre_model_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ def prepare_model_input(
286286
for seq_id in finished_requests_ids:
287287
# ignore requests that are not in the batch, eg. requests
288288
# cancelled while waiting
289-
if idx := self._req_ids2idx.get(seq_id):
289+
if (idx := self._req_ids2idx.get(seq_id)) is not None:
290290
self.model.indices[idx] = False
291291
(input_tokens, input_positions,
292292
input_masks) = self._prepare_decode(seq_group_metadata_list)

0 commit comments

Comments
 (0)