SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

style_doc handles decorators in examples (#15719)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-02-18 14:49:53 +01:00
committed by GitHub
parent ae1f835028
commit d5083c333f
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
utils/style_doc.py
View File

@@ -171,9 +171,14 @@ def format_code_example(code: str, max_len: int, in_docstring: bool = False):
# black may have added some new lines, we remove them # black may have added some new lines, we remove them
code_sample = code_sample.strip() code_sample = code_sample.strip()
in_triple_quotes = False in_triple_quotes = False
in_decorator = False
for line in code_sample.strip().split("\n"): for line in code_sample.strip().split("\n"):
if has_doctest and not is_empty_line(line): if has_doctest and not is_empty_line(line):
prefix = "... " if line.startswith(" ") or line in [")", "]", "}"] or in_triple_quotes else ">>> " prefix = (
"... "
if line.startswith(" ") or line in [")", "]", "}"] or in_triple_quotes or in_decorator
else ">>> "
)
else: else:
prefix = "" prefix = ""
indent_str = "" if is_empty_line(line) else (" " * indent) indent_str = "" if is_empty_line(line) else (" " * indent)
@@ -181,6 +186,10 @@ def format_code_example(code: str, max_len: int, in_docstring: bool = False):
if '"""' in line: if '"""' in line:
in_triple_quotes = not in_triple_quotes in_triple_quotes = not in_triple_quotes
if line.startswith(" "):
in_decorator = False
if line.startswith("@"):
in_decorator = True
formatted_lines.extend([" " * indent + line for line in output.split("\n")]) formatted_lines.extend([" " * indent + line for line in output.split("\n")])
if not output.endswith("===PT-TF-SPLIT==="): if not output.endswith("===PT-TF-SPLIT==="):