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

Ignore empty subfolders when identifying submodules (#15204)

Browse Source 
* Ignore empty subfolders when identifying submodules

* Update utils/check_inits.py
This commit is contained in:
Sylvain Gugger
2022-01-18 09:48:46 -05:00
committed by GitHub
parent 6f0a9b41ef
commit 84c60a7b50
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/check_inits.py
View File

@@ -211,9 +211,13 @@ def get_transformers_submodules():
submodules = [] submodules = []
for path, directories, files in os.walk(PATH_TO_TRANSFORMERS): for path, directories, files in os.walk(PATH_TO_TRANSFORMERS):
for folder in directories: for folder in directories:
# Ignore private modules
if folder.startswith("_"): if folder.startswith("_"):
directories.remove(folder) directories.remove(folder)
continue continue
# Ignore leftovers from branches (empty folders apart from pycache)
if len(list((Path(path) / folder).glob("*.py"))) == 0:
continue
short_path = str((Path(path) / folder).relative_to(PATH_TO_TRANSFORMERS)) short_path = str((Path(path) / folder).relative_to(PATH_TO_TRANSFORMERS))
submodule = short_path.replace(os.path.sep, ".") submodule = short_path.replace(os.path.sep, ".")
submodules.append(submodule) submodules.append(submodule)