File tree Expand file tree Collapse file tree 1 file changed +6
-8
lines changed Expand file tree Collapse file tree 1 file changed +6
-8
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments