Skip to content

Commit b00e69f

Browse files
authored
Fix nits from #20059 (#23548)
Signed-off-by: Harry Mellor <[email protected]>
1 parent 50fede6 commit b00e69f

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

vllm/config/compilation.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,8 @@ class CompilationConfig:
225225
# CudaGraph compilation
226226
cudagraph_mode: Optional[CUDAGraphMode] = None
227227
"""
228-
The mode of the cudagraph.
228+
The mode of the cudagraph:
229+
229230
- NONE, no cudagraph capture.
230231
- PIECEWISE. (v1 default)
231232
- FULL.
@@ -384,13 +385,10 @@ def __repr__(self) -> str:
384385
if pass_config_exclude:
385386
exclude["pass_config"] = pass_config_exclude
386387

387-
# The cast to string is necessary because Pydantic is mocked in docs
388-
# builds and sphinx-argparse doesn't know the return type of decode()
389-
return str(
390-
TypeAdapter(CompilationConfig).dump_json(
391-
self,
392-
exclude=exclude, # type: ignore[arg-type]
393-
exclude_unset=True).decode())
388+
return TypeAdapter(CompilationConfig).dump_json(
389+
self,
390+
exclude=exclude, # type: ignore[arg-type]
391+
exclude_unset=True).decode()
394392

395393
__str__ = __repr__
396394

0 commit comments

Comments
 (0)