Skip to content

GitLab

Unverified Commit 6d917d0e authored by Mark McLoughlin's avatar Mark McLoughlin Committed by GitHub
Browse files

Enable mypy checking on V1 code (#11105) 


Signed-off-by: default avatarMark McLoughlin <markmc@redhat.com>
parent 93abf23a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/v1/worker/gpu_worker.py
View file @ 6d917d0e
...@@ -204,7 +204,7 @@ class Worker: ...@@ -204,7 +204,7 @@ class Worker:
return output if self.rank == 0 else None return output if self.rank == 0 else None
return output return output
def profile(self, is_start=True): def profile(self, is_start: bool = True):
if self.profiler is None: if self.profiler is None:
raise RuntimeError("Profiler is not enabled.") raise RuntimeError("Profiler is not enabled.")
if is_start: if is_start:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment