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

Fix example test and test_fetcher for examples (#16478)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-03-29 12:21:19 -04:00
committed by GitHub
parent 86cff21cf6
commit b62ac4d240
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/tests_fetcher.py
View File

@@ -465,9 +465,9 @@ def infer_tests_to_run(output_file, diff_with_last_commit=False, filters=None):
test_files_to_run.append(f)
# Example files are tested separately
elif f.startswith("examples/pytorch"):
test_files_to_run.append("examples/pytorch/test_examples.py")
test_files_to_run.append("examples/pytorch/test_pytorch_examples.py")
elif f.startswith("examples/flax"):
test_files_to_run.append("examples/flax/test_examples.py")
test_files_to_run.append("examples/flax/test_flax_examples.py")
else:
new_tests = module_to_test_file(f)
if new_tests is not None: