@@ -1210,7 +1210,7 @@ def get_checkpoint_from_config_class(config_class):
12101210 return checkpoint
12111211
12121212
1213- def add_intro_docstring (func , class_name , parent_class = None , indent_level = 0 ):
1213+ def add_intro_docstring (func , class_name , indent_level = 0 ):
12141214 intro_docstring = ""
12151215 if func .__name__ == "forward" :
12161216 intro_docstring = rf"""The [`{ class_name } `] forward method, overrides the `__call__` special method.
@@ -1452,17 +1452,14 @@ def find_sig_line(lines, line_end):
14521452 return sig_line_end
14531453
14541454
1455- def _process_kwargs_parameters (
1456- sig , func , parent_class , model_name_lowercase , documented_kwargs , indent_level , undocumented_parameters
1457- ):
1455+ def _process_kwargs_parameters (sig , func , parent_class , documented_kwargs , indent_level , undocumented_parameters ):
14581456 """
14591457 Process **kwargs parameters if needed.
14601458
14611459 Args:
14621460 sig (`inspect.Signature`): Function signature
14631461 func (`function`): Function the parameters belong to
14641462 parent_class (`class`): Parent class of the function
1465- model_name_lowercase (`str`): Lowercase model name
14661463 documented_kwargs (`dict`): Dictionary of kwargs that are already documented
14671464 indent_level (`int`): Indentation level
14681465 undocumented_parameters (`list`): List to append undocumented parameters to
@@ -1493,7 +1490,7 @@ def _process_kwargs_parameters(
14931490 # Extract documentation for kwargs
14941491 kwargs_documentation = kwarg_param .annotation .__args__ [0 ].__doc__
14951492 if kwargs_documentation is not None :
1496- documented_kwargs , _ = parse_docstring (kwargs_documentation )
1493+ documented_kwargs = parse_docstring (kwargs_documentation )[ 0 ]
14971494
14981495 # Process each kwarg parameter
14991496 for param_name , param_type_annotation in kwarg_param .annotation .__args__ [0 ].__annotations__ .items ():
@@ -1580,7 +1577,7 @@ def _process_parameters_section(
15801577
15811578 # Process **kwargs parameters if needed
15821579 kwargs_docstring = _process_kwargs_parameters (
1583- sig , func , parent_class , model_name_lowercase , documented_kwargs , indent_level , undocumented_parameters
1580+ sig , func , parent_class , documented_kwargs , indent_level , undocumented_parameters
15841581 )
15851582 docstring += kwargs_docstring
15861583
@@ -1740,9 +1737,7 @@ def auto_method_docstring(
17401737 if not docstring .strip ().endswith ("\n " ):
17411738 docstring += "\n "
17421739 else :
1743- docstring = add_intro_docstring (
1744- func , class_name = class_name , parent_class = parent_class , indent_level = indent_level
1745- )
1740+ docstring = add_intro_docstring (func , class_name = class_name , indent_level = indent_level )
17461741
17471742 # Process Parameters section
17481743 docstring += _process_parameters_section (
0 commit comments