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

Fix tests_fetcher (#15376)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-01-27 14:17:48 -05:00
committed by GitHub
parent eab338104d
commit a81fd35524
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
utils/tests_fetcher.py
View File

@@ -197,7 +197,10 @@ def get_test_dependencies(test_fname):
# Tests only have relative imports for other test files # Tests only have relative imports for other test files
relative_imports = re.findall(r"from\s+\.(\S+)\s+import\s+([^\n]+)\n", content) relative_imports = re.findall(r"from\s+\.(\S+)\s+import\s+([^\n]+)\n", content)
relative_imports = [test for test, imp in relative_imports if "# tests_ignore" not in imp] relative_imports = [test for test, imp in relative_imports if "# tests_ignore" not in imp]
return [os.path.join("tests", f"{test}.py") for test in relative_imports] dependencies = [os.path.join("tests", f"{test}.py") for test in relative_imports]
# Some tests use docstrings with relative imports in them and this could catch them, so we check the files
# actually exist
return [f for f in dependencies if os.path.isfile(f)]
def create_reverse_dependency_map(): def create_reverse_dependency_map():