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

Fix pipeline tests env and fetch (#14209)

Browse Source 
* Fix pipeline tests env and fetch

* Fix quality
This commit is contained in:
Sylvain Gugger
2021-10-29 09:35:05 -04:00
committed by GitHub
parent dc540dd316
commit 4ab6a4a086
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
utils/tests_fetcher.py
View File

@@ -264,7 +264,7 @@ SPECIAL_MODULE_TO_TEST_MAP = {
"models/blenderbot_small/tokenization_blenderbot_small.py": "test_tokenization_small_blenderbot.py",
"models/blenderbot_small/tokenization_blenderbot_small_fast.py": "test_tokenization_small_blenderbot.py",
"models/gpt2/modeling_gpt2.py": ["test_modeling_gpt2.py", "test_modeling_megatron_gpt2.py"],
"pipelines/base.py": "test_pipelines_common.py",
"pipelines/base.py": "test_pipelines_*.py",
"pipelines/text2text_generation.py": [
"test_pipelines_text2text_generation.py",
"test_pipelines_summarization.py",
@@ -337,6 +337,7 @@ def module_to_test_file(module_fname):
# launched separately.
EXPECTED_TEST_FILES_NEVER_TOUCHED = [
"tests/test_doc_samples.py", # Doc tests
"tests/test_pipelines_common.py", # Actually checked by the pipeline based file
"tests/sagemaker/test_single_node_gpu.py", # SageMaker test
"tests/sagemaker/test_multi_node_model_parallel.py", # SageMaker test
"tests/sagemaker/test_multi_node_data_parallel.py", # SageMaker test