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

feat: filter try/except when looking at custom code (#21914)

Browse Source 
* feat: filter try/except

* Update src/transformers/dynamic_module_utils.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

---------

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
Zach Nussbaum
2023-03-03 08:43:59 -05:00
committed by GitHub
parent 02a77fa04c
commit c5a1ff9ef0
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/dynamic_module_utils.py
View File

@@ -116,6 +116,9 @@ def check_imports(filename):
with open(filename, "r", encoding="utf-8") as f: with open(filename, "r", encoding="utf-8") as f:
content = f.read() content = f.read()
# filter out try/except block so in custom code we can have try/except imports
content = re.sub(r"\s*try\s*:\s*.*?\s*except\s*:", "", content, flags=re.MULTILINE)
# Imports of the form `import xxx` # Imports of the form `import xxx`
imports = re.findall(r"^\s*import\s+(\S+)\s*$", content, flags=re.MULTILINE) imports = re.findall(r"^\s*import\s+(\S+)\s*$", content, flags=re.MULTILINE)
# Imports of the form `from xxx import yyy` # Imports of the form `from xxx import yyy`