From 0d52f9f582efb82a12e8d9162b43a01b1aa0200f Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Tue, 5 Mar 2024 18:27:25 +0900 Subject: [PATCH] more fix --- tests/models/udop/test_processor_udop.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/models/udop/test_processor_udop.py b/tests/models/udop/test_processor_udop.py index 05855991b1..5e913f8ec6 100644 --- a/tests/models/udop/test_processor_udop.py +++ b/tests/models/udop/test_processor_udop.py @@ -22,7 +22,6 @@ 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, @@ -35,11 +34,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