FIx imports fuyu

huggingface · Aug 13, 2024 · 6d7e086 · 6d7e086
1 parent a72c3cd
commit 6d7e086
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/models/fuyu/test_processing_fuyu.py b/tests/models/fuyu/test_processing_fuyu.py
@@ -4,7 +4,14 @@
 
 import requests
 
-from transformers import AutoTokenizer, is_torch_available, is_vision_available
+from transformers import (
+    AutoProcessor,
+    AutoTokenizer,
+    FuyuImageProcessor,
+    FuyuProcessor,
+    is_torch_available,
+    is_vision_available,
+)
 from transformers.testing_utils import require_torch, require_vision
 
 from ...test_processing_common import ProcessorTesterMixin
@@ -19,9 +26,6 @@
 
     from transformers.models.fuyu.processing_fuyu import construct_full_unpacked_stream, full_unpacked_stream_to_tensor
 
-if is_vision_available() and is_torch_available():
-    from transformers import AutoProcessor, FuyuImageProcessor, FuyuProcessor
-
 
 @require_torch
 @require_vision