make lint

Signed-off-by: Sungjae Lee <33976427+llsj14@users.noreply.github.com>

Author: llsj14

vllm/v1/sample/logits_processor.py

@@ -518,10 +518,20 @@ def is_argmax_invariant(self) -> bool:
     def update_state(self, batch_update: Optional[BatchUpdate]):
         if batch_update:
             for index, params, prompt_tok_ids, output_tok_ids in batch_update.added:
-                max_think_tokens = params.max_think_tokens if isinstance(params, SamplingParams) else None
+                max_think_tokens = (
+                    params.max_think_tokens
+                    if isinstance(params, SamplingParams)
+                    else None
+                )
                 if max_think_tokens is not None:
-                    last_start = self._find_last_token_index(prompt_tok_ids, self.think_start_token_id)
-                    last_end = self._find_last_token_index(prompt_tok_ids, self.think_end_token_id)
+                    last_start = self._find_last_token_index(
+                        prompt_tok_ids,
+                        self.think_start_token_id
+                    )
+                    last_end = self._find_last_token_index(
+                        prompt_tok_ids,
+                        self.think_end_token_id
+                    )
                     in_think = last_start > last_end
                     count = len(prompt_tok_ids) - (last_start + 1) if in_think else 0
 
@@ -534,13 +544,13 @@ def update_state(self, batch_update: Optional[BatchUpdate]):
                     }
 
             for index in batch_update.removed:
-                self._state.pop(index, None)
+                self._state.pop(index, {})
 
             for i1, i2, direction in batch_update.moved:
                 if direction == MoveDirectionality.SWAP:
                     self._state[i1], self._state[i2] = self._state[i2], self._state[i1]
                 else:
-                    self._state[i2] = self._state.pop(i1, None)
+                    self._state[i2] = self._state.pop(i1, {})
 
         # Update in_think and count for all active requests
         for state in self._state.values():