From 06d98e272ee407547f16276fe93d2f3b8bc0ffe4 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Fri, 22 Jul 2022 13:32:19 +0200 Subject: [PATCH] Fix OwlViT tests (#18253) * Fix OwlViT tests * Forgot one --- src/transformers/models/owlvit/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/owlvit/__init__.py b/src/transformers/models/owlvit/__init__.py index 7fbf47124cfc0..8315df69faace 100644 --- a/src/transformers/models/owlvit/__init__.py +++ b/src/transformers/models/owlvit/__init__.py @@ -40,7 +40,7 @@ try: - if not is_vision_available() or not is_torch_available(): + if not is_vision_available(): raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: pass @@ -72,7 +72,7 @@ from .processing_owlvit import OwlViTProcessor try: - if not is_vision_available() or not is_torch_available(): + if not is_vision_available(): raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: pass