Skip to content

Commit d3a033a

Browse files
louis-sheBorda
andcommitted
Apply suggestions from code review
Co-authored-by: Jirka Borovec <[email protected]>
1 parent 08ebfea commit d3a033a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_lightning/utilities/argparse_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def add_argparse_args(cls, parent_parser: ArgumentParser) -> ArgumentParser:
160160

161161
allowed_types = (str, int, float, bool)
162162

163-
args_help = parse_args_from_docstring(cls.__doc__ or cls.__init__.__doc__)
163+
args_help = parse_args_from_docstring(cls.__init__.__doc__ or cls.__doc__)
164164
for arg, arg_types, arg_default in (
165165
at for at in get_init_arguments_and_types(cls) if at[0] not in depr_arg_names
166166
):
@@ -226,7 +226,7 @@ def parse_args_from_docstring(docstring: str) -> Dict[str, str]:
226226
current_arg, arg_description = stripped.split(':', maxsplit=1)
227227
parsed[current_arg] = arg_description.lstrip()
228228
elif line_indent > arg_block_indent:
229-
parsed[current_arg] = f'{parsed[current_arg]} {stripped}'
229+
parsed[current_arg] += f' {stripped}'
230230
return parsed
231231

232232

0 commit comments

Comments
 (0)