From 6d7e086acdb9bd4f8fd4ea03a3615c6b8e212519 Mon Sep 17 00:00:00 2001
From: yonigozlan <yoni.gozlan@huggingface.co>
Date: Tue, 13 Aug 2024 21:03:56 +0000
Subject: [PATCH] FIx imports fuyu

---
 tests/models/fuyu/test_processing_fuyu.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/tests/models/fuyu/test_processing_fuyu.py b/tests/models/fuyu/test_processing_fuyu.py
index 8ccfad4328cf9e..ed0a8d5e462d1f 100644
--- 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