Fix pipeline tests (#18487)

* Fix pipeline tests

* Make sure all pipelines tests run with init changes
This commit is contained in:
Sylvain Gugger
2022-08-05 09:14:51 -04:00
committed by GitHub
parent c7849d9efc
commit 70fa1a8d26
3 changed files with 4 additions and 2 deletions

View File

@@ -377,6 +377,7 @@ SPECIAL_MODULE_TO_TEST_MAP = {
],
"optimization.py": "optimization/test_optimization.py",
"optimization_tf.py": "optimization/test_optimization_tf.py",
"pipelines/__init__.py": "pipelines/test_pipelines_*.py",
"pipelines/base.py": "pipelines/test_pipelines_*.py",
"pipelines/text2text_generation.py": [
"pipelines/test_pipelines_text2text_generation.py",