diff --git a/vllm/worker/worker_base.py b/vllm/worker/worker_base.py index 0b37caa71669..200026dc7282 100644 --- a/vllm/worker/worker_base.py +++ b/vllm/worker/worker_base.py @@ -202,8 +202,7 @@ def remove_lora(self, lora_id: int) -> bool: raise ValueError(f"{type(self)} does not support LoRA") def pin_lora(self, lora_id: int) -> bool: - return ValueError( - f"{type(self)} does not support LoRA") # type: ignore + raise ValueError(f"{type(self)} does not support LoRA") def list_loras(self) -> Set[int]: raise ValueError(f"{type(self)} does not support LoRA") @@ -398,7 +397,7 @@ def execute_model( model_input, worker_input, kwargs = inputs num_steps = worker_input.num_steps - if (execute_model_req is not None and execute_model_req.spec_step_idx): + if execute_model_req is not None and execute_model_req.spec_step_idx: kwargs["spec_step_idx"] = execute_model_req.spec_step_idx self.execute_worker(worker_input)