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