@@ -1227,7 +1227,7 @@ def get_checkpoint_from_config_class(config_class):
12271227 return checkpoint
12281228
12291229
1230- def add_intro_docstring (func , class_name , parent_class = None , indent_level = 0 ):
1230+ def add_intro_docstring (func , class_name , indent_level = 0 ):
12311231 intro_docstring = ""
12321232 if func .__name__ == "forward" :
12331233 intro_docstring = rf"""The [`{ class_name } `] forward method, overrides the `__call__` special method.
@@ -1469,17 +1469,14 @@ def find_sig_line(lines, line_end):
14691469 return sig_line_end
14701470
14711471
1472- def _process_kwargs_parameters (
1473- sig , func , parent_class , model_name_lowercase , documented_kwargs , indent_level , undocumented_parameters
1474- ):
1472+ def _process_kwargs_parameters (sig , func , parent_class , documented_kwargs , indent_level , undocumented_parameters ):
14751473 """
14761474 Process **kwargs parameters if needed.
14771475
14781476 Args:
14791477 sig (`inspect.Signature`): Function signature
14801478 func (`function`): Function the parameters belong to
14811479 parent_class (`class`): Parent class of the function
1482- model_name_lowercase (`str`): Lowercase model name
14831480 documented_kwargs (`dict`): Dictionary of kwargs that are already documented
14841481 indent_level (`int`): Indentation level
14851482 undocumented_parameters (`list`): List to append undocumented parameters to
@@ -1510,7 +1507,7 @@ def _process_kwargs_parameters(
15101507 # Extract documentation for kwargs
15111508 kwargs_documentation = kwarg_param .annotation .__args__ [0 ].__doc__
15121509 if kwargs_documentation is not None :
1513- documented_kwargs , _ = parse_docstring (kwargs_documentation )
1510+ documented_kwargs = parse_docstring (kwargs_documentation )[ 0 ]
15141511
15151512 # Process each kwarg parameter
15161513 for param_name , param_type_annotation in kwarg_param .annotation .__args__ [0 ].__annotations__ .items ():
@@ -1597,7 +1594,7 @@ def _process_parameters_section(
15971594
15981595 # Process **kwargs parameters if needed
15991596 kwargs_docstring = _process_kwargs_parameters (
1600- sig , func , parent_class , model_name_lowercase , documented_kwargs , indent_level , undocumented_parameters
1597+ sig , func , parent_class , documented_kwargs , indent_level , undocumented_parameters
16011598 )
16021599 docstring += kwargs_docstring
16031600
@@ -1757,9 +1754,7 @@ def auto_method_docstring(
17571754 if not docstring .strip ().endswith ("\n " ):
17581755 docstring += "\n "
17591756 else :
1760- docstring = add_intro_docstring (
1761- func , class_name = class_name , parent_class = parent_class , indent_level = indent_level
1762- )
1757+ docstring = add_intro_docstring (func , class_name = class_name , indent_level = indent_level )
17631758
17641759 # Process Parameters section
17651760 docstring += _process_parameters_section (
0 commit comments