From 22d41bbd54c072598c884185a9c8ef058002b828 Mon Sep 17 00:00:00 2001 From: zhangyue66 Date: Wed, 22 Apr 2026 11:53:10 +0800 Subject: [PATCH 1/3] update --- .../models/paddleocr_vl/test_modeling_paddleocr_vl.py | 3 ++- .../pp_chart2table/test_modeling_pp_chart2table.py | 3 ++- .../pp_doclayout_v2/test_modeling_pp_doclayout_v2.py | 8 ++++++-- .../pp_doclayout_v3/test_modeling_pp_doclayout_v3.py | 11 ++++++----- tests/models/pp_lcnet/test_modeling_pp_lcnet.py | 11 ++++++----- .../test_modeling_pp_ocrv5_mobile_det.py | 8 ++++++-- .../test_modeling_pp_ocrv5_mobile_rec.py | 11 ++++++----- .../test_modeling_pp_ocrv5_server_det.py | 11 ++++++----- .../test_modeling_pp_ocrv5_server_rec.py | 11 ++++++----- tests/models/slanext/test_modeling_slanext.py | 8 ++++++-- tests/models/uvdoc/test_modeling_uvdoc.py | 9 +++++---- utils/fetch_hub_objects_for_ci.py | 8 ++++++++ 12 files changed, 65 insertions(+), 37 deletions(-) diff --git a/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py b/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py index 1e95ea4741f4..090425998609 100644 --- a/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py +++ b/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py @@ -43,6 +43,7 @@ ids_tensor, ) from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -360,7 +361,7 @@ def setUp(self): "content": [ { "type": "image", - "url": "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/ocr_demo2.jpg", + "url": url_to_local_path("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/ocr_demo2.jpg"), }, {"type": "text", "text": "OCR:"}, ], diff --git a/tests/models/pp_chart2table/test_modeling_pp_chart2table.py b/tests/models/pp_chart2table/test_modeling_pp_chart2table.py index b573723c4d13..0cb57cba6ae7 100644 --- a/tests/models/pp_chart2table/test_modeling_pp_chart2table.py +++ b/tests/models/pp_chart2table/test_modeling_pp_chart2table.py @@ -17,6 +17,7 @@ from transformers import AutoModelForImageTextToText, AutoProcessor from transformers.testing_utils import cleanup, require_torch, require_vision, slow, torch_device +from ...test_processing_common import url_to_local_path @slow @@ -33,7 +34,7 @@ def setUp(self): "content": [ { "type": "image", - "url": "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/chart_parsing_02.png", + "url": url_to_local_path("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/chart_parsing_02.png"), }, ], }, diff --git a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py index 941e676bb1fd..4b2b60cb73ef 100644 --- a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py +++ b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py @@ -21,6 +21,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( PPDocLayoutV2Config, @@ -41,6 +42,7 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -583,8 +585,10 @@ def setUp(self): model_path = "PaddlePaddle/PP-DocLayoutV2_safetensors" self.model = PPDocLayoutV2ForObjectDetection.from_pretrained(model_path).to(torch_device) self.image_processor = PPDocLayoutV2ImageProcessor.from_pretrained(model_path) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/layout_demo.jpg" - self.image = Image.open(requests.get(url, stream=True).raw) + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/layout_demo.jpg" + ) + self.image = load_image(img_url) def tearDown(self): cleanup(torch_device, gc_collect=True) diff --git a/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py b/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py index f20518bfb0fc..94a8f01e306f 100644 --- a/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py +++ b/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py @@ -20,6 +20,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( PPDocLayoutV3Config, @@ -39,14 +40,12 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): import torch -if is_vision_available(): - from PIL import Image - class PPDocLayoutV3ModelTester: def __init__( @@ -457,8 +456,10 @@ def setUp(self): self.image_processor = ( PPDocLayoutV3ImageProcessor.from_pretrained(model_path) if is_vision_available() else None ) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/layout_demo.jpg" - self.image = Image.open(requests.get(url, stream=True).raw) + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/layout_demo.jpg" + ) + self.image = load_image(img_url) def test_inference_object_detection_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/pp_lcnet/test_modeling_pp_lcnet.py b/tests/models/pp_lcnet/test_modeling_pp_lcnet.py index c7f31f43129e..37f0627059af 100644 --- a/tests/models/pp_lcnet/test_modeling_pp_lcnet.py +++ b/tests/models/pp_lcnet/test_modeling_pp_lcnet.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( PPLCNetBackbone, @@ -40,14 +41,12 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): import torch -if is_vision_available(): - from PIL import Image - class PPLCNetModelTester: def __init__( @@ -282,8 +281,10 @@ def setUp(self): model_path = "PaddlePaddle/PP-LCNet_x1_0_doc_ori_safetensors" self.model = PPLCNetForImageClassification.from_pretrained(model_path).to(torch_device) self.image_processor = PPLCNetImageProcessor.from_pretrained(model_path) if is_vision_available() else None - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/img_rot180_demo.jpg" - self.image = Image.open(requests.get(url, stream=True).raw) + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/img_rot180_demo.jpg" + ) + self.image = load_image(img_url) def test_inference_image_classification_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py index a26243fb416d..c253e05d9b5a 100644 --- a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py +++ b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( PPOCRV5MobileDetConfig, @@ -38,6 +39,7 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -242,8 +244,10 @@ def setUp(self): self.image_processor = ( PPOCRV5ServerDetImageProcessor.from_pretrained(model_path) if is_vision_available() else None ) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png" - self.image = Image.open(requests.get(url, stream=True).raw) + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png" + ) + self.image = load_image(img_url) def test_inference_object_detection_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py b/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py index dc530860d3a5..d972885df07d 100644 --- a/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py +++ b/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -39,14 +40,12 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): import torch -if is_vision_available(): - from PIL import Image - class PPOCRV5MobileRecModelTester: def __init__( @@ -245,8 +244,10 @@ def setUp(self): self.image_processor = ( AutoImageProcessor.from_pretrained(model_path, return_tensors="pt") if is_vision_available() else None ) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png" - self.image = Image.open(requests.get(url, stream=True).raw).convert("RGB") + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png" + ) + self.image = load_image(img_url) def test_inference_text_recognition_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py index 4f41dbb7a31f..4a7ef71f32b9 100644 --- a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py +++ b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( PPOCRV5ServerDetConfig, @@ -40,6 +41,7 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -263,11 +265,10 @@ def setUp(self): self.image_processor = ( PPOCRV5ServerDetImageProcessor.from_pretrained(model_path) if is_vision_available() else None ) - self.image = Image.open( - requests.get( - "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png", stream=True - ).raw - ).convert("RGB") + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png" + ) + self.image = load_image(img_url) def test_inference_object_detection_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py b/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py index e23e28b65c8b..9afc971e4b2a 100644 --- a/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py +++ b/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -39,14 +40,12 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): import torch -if is_vision_available(): - from PIL import Image - class PPOCRV5ServerRecModelTester: def __init__( @@ -251,8 +250,10 @@ def setUp(self): self.image_processor = ( AutoImageProcessor.from_pretrained(model_path, return_tensors="pt") if is_vision_available() else None ) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png" - self.image = Image.open(requests.get(url, stream=True).raw).convert("RGB") + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png" + ) + self.image = load_image(img_url) def test_inference_text_recognition_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/slanext/test_modeling_slanext.py b/tests/models/slanext/test_modeling_slanext.py index 759073a5ec6b..18c2e65ee2f0 100644 --- a/tests/models/slanext/test_modeling_slanext.py +++ b/tests/models/slanext/test_modeling_slanext.py @@ -21,6 +21,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -41,6 +42,7 @@ from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor from ...test_pipeline_mixin import PipelineTesterMixin +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -295,8 +297,10 @@ def setUp(self): model_path = "PaddlePaddle/SLANeXt_wired_safetensors" self.model = AutoModelForTableRecognition.from_pretrained(model_path, dtype=torch.float32).to(torch_device) self.image_processor = AutoImageProcessor.from_pretrained(model_path) - url = "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg" - self.image = Image.open(requests.get(url, stream=True).raw) + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg" + ) + self.image = load_image(img_url) def test_inference_table_recognition_head(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/tests/models/uvdoc/test_modeling_uvdoc.py b/tests/models/uvdoc/test_modeling_uvdoc.py index 4b4c6ae960ca..40a14690f7be 100644 --- a/tests/models/uvdoc/test_modeling_uvdoc.py +++ b/tests/models/uvdoc/test_modeling_uvdoc.py @@ -19,6 +19,7 @@ import requests from parameterized import parameterized +from transformers.image_utils import load_image from transformers import ( AutoModel, @@ -41,6 +42,7 @@ from ...test_backbone_common import BackboneTesterMixin from ...test_configuration_common import ConfigTester from ...test_modeling_common import ModelTesterMixin, floats_tensor +from ...test_processing_common import url_to_local_path if is_torch_available(): @@ -310,11 +312,10 @@ def setUp(self): model_path = "PaddlePaddle/UVDoc_safetensors" self.model = AutoModel.from_pretrained(model_path).to(torch_device) self.image_processor = UVDocImageProcessor() - self.image = Image.open( - requests.get( - "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/doc_test.jpg", stream=True - ).raw + img_url = url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/doc_test.jpg" ) + self.image = load_image(img_url) def test_inference_document_rectification(self): inputs = self.image_processor(images=self.image, return_tensors="pt").to(torch_device) diff --git a/utils/fetch_hub_objects_for_ci.py b/utils/fetch_hub_objects_for_ci.py index 3d229637df70..0869847a8518 100644 --- a/utils/fetch_hub_objects_for_ci.py +++ b/utils/fetch_hub_objects_for_ci.py @@ -39,6 +39,14 @@ URLS_FOR_TESTING_DATA = [ # TODO: copy those to our hf-internal-testing dataset and fix all tests using them + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/ocr_demo2.jpg", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/doc_test.jpg", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/chart_parsing_02.png", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/layout_demo.jpg", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/img_rot180_demo.jpg", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png", + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg", "http://images.cocodataset.org/val2017/000000000139.jpg", "http://images.cocodataset.org/val2017/000000000285.jpg", "http://images.cocodataset.org/val2017/000000000632.jpg", From ddedc78b0332602a6fe72fa586755733afc2159c Mon Sep 17 00:00:00 2001 From: zhangyue66 Date: Wed, 22 Apr 2026 11:54:42 +0800 Subject: [PATCH 2/3] fix style --- tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py | 4 +++- .../pp_chart2table/test_modeling_pp_chart2table.py | 5 ++++- .../pp_doclayout_v2/test_modeling_pp_doclayout_v2.py | 5 ++--- .../pp_doclayout_v3/test_modeling_pp_doclayout_v3.py | 3 +-- tests/models/pp_lcnet/test_modeling_pp_lcnet.py | 3 +-- .../test_modeling_pp_ocrv5_mobile_det.py | 5 ++--- .../test_modeling_pp_ocrv5_mobile_rec.py | 3 +-- .../test_modeling_pp_ocrv5_server_det.py | 5 ++--- .../test_modeling_pp_ocrv5_server_rec.py | 3 +-- tests/models/slanext/test_modeling_slanext.py | 5 ++--- tests/models/uvdoc/test_modeling_uvdoc.py | 9 +++------ 11 files changed, 22 insertions(+), 28 deletions(-) diff --git a/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py b/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py index 090425998609..7ccd2056bff7 100644 --- a/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py +++ b/tests/models/paddleocr_vl/test_modeling_paddleocr_vl.py @@ -361,7 +361,9 @@ def setUp(self): "content": [ { "type": "image", - "url": url_to_local_path("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/ocr_demo2.jpg"), + "url": url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/ocr_demo2.jpg" + ), }, {"type": "text", "text": "OCR:"}, ], diff --git a/tests/models/pp_chart2table/test_modeling_pp_chart2table.py b/tests/models/pp_chart2table/test_modeling_pp_chart2table.py index 0cb57cba6ae7..799da10492ea 100644 --- a/tests/models/pp_chart2table/test_modeling_pp_chart2table.py +++ b/tests/models/pp_chart2table/test_modeling_pp_chart2table.py @@ -17,6 +17,7 @@ from transformers import AutoModelForImageTextToText, AutoProcessor from transformers.testing_utils import cleanup, require_torch, require_vision, slow, torch_device + from ...test_processing_common import url_to_local_path @@ -34,7 +35,9 @@ def setUp(self): "content": [ { "type": "image", - "url": url_to_local_path("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/chart_parsing_02.png"), + "url": url_to_local_path( + "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/chart_parsing_02.png" + ), }, ], }, diff --git a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py index 4b2b60cb73ef..892a4f3acd06 100644 --- a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py +++ b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py @@ -19,9 +19,7 @@ import tempfile import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( PPDocLayoutV2Config, @@ -30,6 +28,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( cleanup, require_torch, @@ -49,7 +48,7 @@ import torch if is_vision_available(): - from PIL import Image + pass class PPDocLayoutV2ModelTester: diff --git a/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py b/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py index 94a8f01e306f..d44e01dce40d 100644 --- a/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py +++ b/tests/models/pp_doclayout_v3/test_modeling_pp_doclayout_v3.py @@ -18,9 +18,7 @@ import math import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( PPDocLayoutV3Config, @@ -29,6 +27,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, diff --git a/tests/models/pp_lcnet/test_modeling_pp_lcnet.py b/tests/models/pp_lcnet/test_modeling_pp_lcnet.py index 37f0627059af..f404d682dd40 100644 --- a/tests/models/pp_lcnet/test_modeling_pp_lcnet.py +++ b/tests/models/pp_lcnet/test_modeling_pp_lcnet.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( PPLCNetBackbone, @@ -29,6 +27,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, diff --git a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py index c253e05d9b5a..e473a2992d4a 100644 --- a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py +++ b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( PPOCRV5MobileDetConfig, @@ -28,6 +26,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_cv2, require_torch, @@ -46,7 +45,7 @@ import torch if is_vision_available(): - from PIL import Image + pass class PPOCRV5MobileDetModelTester: diff --git a/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py b/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py index d972885df07d..5d91b67a6412 100644 --- a/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py +++ b/tests/models/pp_ocrv5_mobile_rec/test_modeling_pp_ocrv5_mobile_rec.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -29,6 +27,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, diff --git a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py index 4a7ef71f32b9..2a00061b1915 100644 --- a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py +++ b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( PPOCRV5ServerDetConfig, @@ -29,6 +27,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_cv2, require_torch, @@ -48,7 +47,7 @@ import torch if is_vision_available(): - from PIL import Image + pass class PPOCRV5ServerDetModelTester: diff --git a/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py b/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py index 9afc971e4b2a..c6efa3877228 100644 --- a/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py +++ b/tests/models/pp_ocrv5_server_rec/test_modeling_pp_ocrv5_server_rec.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -29,6 +27,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, diff --git a/tests/models/slanext/test_modeling_slanext.py b/tests/models/slanext/test_modeling_slanext.py index 18c2e65ee2f0..a36f292f21dc 100644 --- a/tests/models/slanext/test_modeling_slanext.py +++ b/tests/models/slanext/test_modeling_slanext.py @@ -19,9 +19,7 @@ import tempfile import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( AutoImageProcessor, @@ -31,6 +29,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, @@ -49,7 +48,7 @@ import torch if is_vision_available(): - from PIL import Image + pass class SLANeXtModelTester: diff --git a/tests/models/uvdoc/test_modeling_uvdoc.py b/tests/models/uvdoc/test_modeling_uvdoc.py index 40a14690f7be..5c2c5bdbbf44 100644 --- a/tests/models/uvdoc/test_modeling_uvdoc.py +++ b/tests/models/uvdoc/test_modeling_uvdoc.py @@ -17,9 +17,7 @@ import inspect import unittest -import requests from parameterized import parameterized -from transformers.image_utils import load_image from transformers import ( AutoModel, @@ -31,6 +29,7 @@ is_torch_available, is_vision_available, ) +from transformers.image_utils import load_image from transformers.testing_utils import ( require_torch, require_torch_accelerator, @@ -49,7 +48,7 @@ import torch if is_vision_available(): - from PIL import Image + pass class UVDocModelTester: @@ -312,9 +311,7 @@ def setUp(self): model_path = "PaddlePaddle/UVDoc_safetensors" self.model = AutoModel.from_pretrained(model_path).to(torch_device) self.image_processor = UVDocImageProcessor() - img_url = url_to_local_path( - "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/doc_test.jpg" - ) + img_url = url_to_local_path("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/doc_test.jpg") self.image = load_image(img_url) def test_inference_document_rectification(self): From 5434ad9c816ba2f2f5b3838a4dd40026c2d7037b Mon Sep 17 00:00:00 2001 From: zhangyue66 Date: Wed, 22 Apr 2026 11:58:40 +0800 Subject: [PATCH 3/3] remove is_vision_available --- tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py | 4 ---- .../pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py | 3 --- .../pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py | 3 --- tests/models/slanext/test_modeling_slanext.py | 4 ---- tests/models/uvdoc/test_modeling_uvdoc.py | 4 ---- 5 files changed, 18 deletions(-) diff --git a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py index 892a4f3acd06..c2b4b3cf0d66 100644 --- a/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py +++ b/tests/models/pp_doclayout_v2/test_modeling_pp_doclayout_v2.py @@ -26,7 +26,6 @@ PPDocLayoutV2ForObjectDetection, PPDocLayoutV2ImageProcessor, is_torch_available, - is_vision_available, ) from transformers.image_utils import load_image from transformers.testing_utils import ( @@ -47,9 +46,6 @@ if is_torch_available(): import torch -if is_vision_available(): - pass - class PPDocLayoutV2ModelTester: def __init__( diff --git a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py index e473a2992d4a..2f044d78eab2 100644 --- a/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py +++ b/tests/models/pp_ocrv5_mobile_det/test_modeling_pp_ocrv5_mobile_det.py @@ -44,9 +44,6 @@ if is_torch_available(): import torch -if is_vision_available(): - pass - class PPOCRV5MobileDetModelTester: def __init__( diff --git a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py index 2a00061b1915..9ce449d1f602 100644 --- a/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py +++ b/tests/models/pp_ocrv5_server_det/test_modeling_pp_ocrv5_server_det.py @@ -46,9 +46,6 @@ if is_torch_available(): import torch -if is_vision_available(): - pass - class PPOCRV5ServerDetModelTester: def __init__( diff --git a/tests/models/slanext/test_modeling_slanext.py b/tests/models/slanext/test_modeling_slanext.py index a36f292f21dc..1de896fe27e3 100644 --- a/tests/models/slanext/test_modeling_slanext.py +++ b/tests/models/slanext/test_modeling_slanext.py @@ -27,7 +27,6 @@ SLANeXtConfig, SLANeXtForTableRecognition, is_torch_available, - is_vision_available, ) from transformers.image_utils import load_image from transformers.testing_utils import ( @@ -47,9 +46,6 @@ if is_torch_available(): import torch -if is_vision_available(): - pass - class SLANeXtModelTester: def __init__( diff --git a/tests/models/uvdoc/test_modeling_uvdoc.py b/tests/models/uvdoc/test_modeling_uvdoc.py index 5c2c5bdbbf44..358593a0d3c1 100644 --- a/tests/models/uvdoc/test_modeling_uvdoc.py +++ b/tests/models/uvdoc/test_modeling_uvdoc.py @@ -27,7 +27,6 @@ UVDocImageProcessor, UVDocModel, is_torch_available, - is_vision_available, ) from transformers.image_utils import load_image from transformers.testing_utils import ( @@ -47,9 +46,6 @@ if is_torch_available(): import torch -if is_vision_available(): - pass - class UVDocModelTester: def __init__(