From 460447b284957f8990466b9590eb2ba0adebadf2 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Mon, 31 Jul 2023 11:31:40 +0200 Subject: [PATCH] fix --- src/transformers/utils/doc.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/transformers/utils/doc.py b/src/transformers/utils/doc.py index f5eea7ae4e11..17aeadcfdf99 100644 --- a/src/transformers/utils/doc.py +++ b/src/transformers/utils/doc.py @@ -122,6 +122,11 @@ def _prepare_output_docstrings(output_type, config_class, min_indent=None): if i < len(lines): params_docstring = "\n".join(lines[(i + 1) :]) params_docstring = _convert_output_args_doc(params_docstring) + else: + raise ValueError( + f"No `Args` or `Parameters` section is found in the docstring of `{output_type.__name__}`. Make sure it has" + "docstring and contain either `Args` or `Parameters`." + ) # Add the return introduction full_output_type = f"{output_type.__module__}.{output_type.__name__}"