[Tests] Improve vision tests (#14458)

* Improve tests * Install vision for tf tests
2021-11-24 15:22:20 +01:00
parent f2e90bcb8f
commit 3772af49ce
6 changed files with 9 additions and 5 deletions
--- a/tests/test_modeling_vit.py
+++ b/tests/test_modeling_vit.py
@@ -331,6 +331,7 @@ def prepare_img():
    return image


+@require_torch
@require_vision
 class ViTModelIntegrationTest(unittest.TestCase):
    @cached_property