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

Fix expected value in tests of the test fetcher (#24077)

Browse Source 
* Fix expected value in tests of the test fetcher

* Fix trigger for repo util tests
This commit is contained in:
Sylvain Gugger
2023-06-07 11:38:56 -04:00
committed by GitHub
parent 5c9394b54c
commit 89b00eef94
2 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/repo_utils/test_tests_fetcher.py
View File

@@ -526,10 +526,17 @@ src/transformers/configuration_utils.py
create_tmp_repo(tmp_folder) create_tmp_repo(tmp_folder)
expected_example_deps = { expected_example_deps = {
"examples/flax/test_flax_examples.py": ["examples/flax/text-classification/run_glue.py"], "examples/flax/test_flax_examples.py": [
"examples/pytorch/test_pytorch_examples.py": ["examples/pytorch/text-classification/run_glue.py"], "examples/flax/text-classification/run_glue.py",
"examples/flax/test_flax_examples.py",
],
"examples/pytorch/test_pytorch_examples.py": [
"examples/pytorch/text-classification/run_glue.py",
"examples/pytorch/test_pytorch_examples.py",
],
"examples/tensorflow/test_tensorflow_examples.py": [ "examples/tensorflow/test_tensorflow_examples.py": [
"examples/tensorflow/text-classification/run_glue.py" "examples/tensorflow/text-classification/run_glue.py",
"examples/tensorflow/test_tensorflow_examples.py",
], ],
} }

8
utils/tests_fetcher.py
View File

@@ -353,7 +353,9 @@ def extract_imports(module_fname, cache=None):
content = f.read() content = f.read()
# Filter out all docstrings to not get imports in code examples. # Filter out all docstrings to not get imports in code examples.
splits = content.split('"""') # fmt: off
splits = content.split('\"\"\"')
# fmt: on
content = "".join(splits[::2]) content = "".join(splits[::2])
module_parts = str(module_fname).split(os.path.sep) module_parts = str(module_fname).split(os.path.sep)
@@ -694,12 +696,14 @@ def infer_tests_to_run(output_file, diff_with_last_commit=False, filter_models=T
if f in test_map: if f in test_map:
test_files_to_run.extend(test_map[f]) test_files_to_run.extend(test_map[f])
test_files_to_run = sorted(set(test_files_to_run)) test_files_to_run = sorted(set(test_files_to_run))
# Remove repo utils tests
test_files_to_run = [f for f in test_files_to_run if not f.split(os.path.sep)[1] == "repo_utils"]
# Remove SageMaker tests # Remove SageMaker tests
test_files_to_run = [f for f in test_files_to_run if not f.split(os.path.sep)[1] == "sagemaker"] test_files_to_run = [f for f in test_files_to_run if not f.split(os.path.sep)[1] == "sagemaker"]
# Make sure we did not end up with a test file that was removed # Make sure we did not end up with a test file that was removed
test_files_to_run = [f for f in test_files_to_run if (PATH_TO_REPO / f).exists()] test_files_to_run = [f for f in test_files_to_run if (PATH_TO_REPO / f).exists()]
repo_utils_launch = any(f.split(os.path.sep)[1] == "repo_utils" for f in modified_files) repo_utils_launch = any(f.split(os.path.sep)[0] == "utils" for f in modified_files)
if repo_utils_launch: if repo_utils_launch:
repo_util_file = Path(output_file).parent / "test_repo_utils.txt" repo_util_file = Path(output_file).parent / "test_repo_utils.txt"