tests/test_pipelines_common.py · 6b586ed18c137a152c0f77af2d21fee346b4c760 · chenpangpang / transformers

Move `image-classification` pipeline to new testing (#13272) · 6b586ed1

Nicolas Patry authored Aug 26, 2021

- Enforce `test_small_models_{tf,pt}` methods to exist (enforce checking
actual values in small tests)
- Add support for non RGB image for the pipeline.

6b586ed1

test_pipelines_common.py 15.9 KB

Replace test_pipelines_common.py