riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Update test fetcher when we want to test all (#35364)

Browse Source 
* [test-all]

* style

* [test-all]

* [test_all]

* [test_all]

* style
This commit is contained in:
Arthur 2024-12-20 15:10:43 +01:00 committed by GitHub
parent 34ad1bd287
commit 6fae2a84ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
utils/tests_fetcher.py
View File

@ -995,9 +995,7 @@ def _print_list(l) -> str:
def infer_tests_to_run(
output_file: str,
diff_with_last_commit: bool = False,
filter_models: bool = False,
output_file: str, diff_with_last_commit: bool = False, filter_models: bool = False, test_all: bool = False
):
"""
The main function called by the test fetcher. Determines the tests to run from the diff.
@ -1018,7 +1016,11 @@ def infer_tests_to_run(
Whether or not to filter the tests to core models only, when a file modified results in a lot of model
tests.
"""
modified_files = get_modified_python_files(diff_with_last_commit=diff_with_last_commit)
if not test_all:
modified_files = get_modified_python_files(diff_with_last_commit=diff_with_last_commit)
else:
modified_files = [str(k) for k in PATH_TO_TESTS.glob("*/*") if str(k).endswith(".py") and "test_" in str(k)]
print("\n### test_all is TRUE, FETCHING ALL FILES###\n")
print(f"\n### MODIFIED FILES ###\n{_print_list(modified_files)}")
# Create the map that will give us all impacted modules.
@ -1230,5 +1232,6 @@ if __name__ == "__main__":
args.output_file,
diff_with_last_commit=diff_with_last_commit,
filter_models=False,
test_all=commit_flags["test_all"],
)
filter_tests(args.output_file, ["repo_utils"])