Unverified Commit b00e69f8 authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Fix nits from #20059 (#23548)


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent 50fede66
...@@ -225,7 +225,8 @@ class CompilationConfig: ...@@ -225,7 +225,8 @@ class CompilationConfig:
# CudaGraph compilation # CudaGraph compilation
cudagraph_mode: Optional[CUDAGraphMode] = None cudagraph_mode: Optional[CUDAGraphMode] = None
""" """
The mode of the cudagraph. The mode of the cudagraph:
- NONE, no cudagraph capture. - NONE, no cudagraph capture.
- PIECEWISE. (v1 default) - PIECEWISE. (v1 default)
- FULL. - FULL.
...@@ -384,13 +385,10 @@ class CompilationConfig: ...@@ -384,13 +385,10 @@ class CompilationConfig:
if pass_config_exclude: if pass_config_exclude:
exclude["pass_config"] = pass_config_exclude exclude["pass_config"] = pass_config_exclude
# The cast to string is necessary because Pydantic is mocked in docs return TypeAdapter(CompilationConfig).dump_json(
# builds and sphinx-argparse doesn't know the return type of decode() self,
return str( exclude=exclude, # type: ignore[arg-type]
TypeAdapter(CompilationConfig).dump_json( exclude_unset=True).decode()
self,
exclude=exclude, # type: ignore[arg-type]
exclude_unset=True).decode())
__str__ = __repr__ __str__ = __repr__
......
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