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

Check copies match full class/function names (#10030)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-02-08 04:58:25 -05:00
committed by GitHub
parent d51302cca0
commit 04fd783cc5
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
utils/check_copies.py
View File

@@ -36,6 +36,7 @@ def find_code_in_transformers(object_name):
module = parts[i] module = parts[i]
while i < len(parts) and not os.path.isfile(os.path.join(TRANSFORMERS_PATH, f"{module}.py")): while i < len(parts) and not os.path.isfile(os.path.join(TRANSFORMERS_PATH, f"{module}.py")):
i += 1 i += 1
if i < len(parts):
module = os.path.join(module, parts[i]) module = os.path.join(module, parts[i])
if i >= len(parts): if i >= len(parts):
raise ValueError( raise ValueError(
@@ -49,7 +50,9 @@ def find_code_in_transformers(object_name):
indent = "" indent = ""
line_index = 0 line_index = 0
for name in parts[i + 1 :]: for name in parts[i + 1 :]:
while line_index < len(lines) and re.search(fr"^{indent}(class|def)\s+{name}", lines[line_index]) is None: while (
line_index < len(lines) and re.search(fr"^{indent}(class|def)\s+{name}(\(|\:)", lines[line_index]) is None
):
line_index += 1 line_index += 1
indent += " " indent += " "
line_index += 1 line_index += 1