From 57d007b912604ac3fac4dd60cd1e6f43908fba08 Mon Sep 17 00:00:00 2001
From: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Date: Tue, 5 Mar 2024 10:39:42 +0100
Subject: [PATCH] Revert-commit 0d52f9f582efb82a12e8d9162b43a01b1aa0200f
 (#29455)

* style

* revert with RP

* nit

* exact revert
---
 tests/models/udop/test_processor_udop.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/models/udop/test_processor_udop.py b/tests/models/udop/test_processor_udop.py
index 5e913f8ec6..05855991b1 100644
--- a/tests/models/udop/test_processor_udop.py
+++ b/tests/models/udop/test_processor_udop.py
@@ -22,6 +22,7 @@ from typing import List
 import numpy as np
 
 from transformers import PreTrainedTokenizer, PreTrainedTokenizerBase, PreTrainedTokenizerFast
+from transformers.models.udop import UdopTokenizer, UdopTokenizerFast
 from transformers.testing_utils import (
     require_pytesseract,
     require_sentencepiece,
@@ -34,11 +35,11 @@ from transformers.utils import FEATURE_EXTRACTOR_NAME, cached_property, is_pytes
 
 if is_torch_available():
     import torch
-    from transformers import UdopTokenizer, UdopTokenizerFast
 
 
 if is_pytesseract_available():
     from PIL import Image
+
     from transformers import LayoutLMv3ImageProcessor, UdopProcessor