riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Update test_image_processing_pix2struct.py (#22543)

Browse Source
This commit is contained in:
Younes Belkada 2023-04-03 21:26:35 +02:00 committed by GitHub
parent c14d31294e
commit 159ff3342c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/pix2struct/test_image_processing_pix2struct.py
View File

@ -65,7 +65,7 @@ class Pix2StructImageProcessingTester(unittest.TestCase):
return {"do_normalize": self.do_normalize, "do_convert_rgb": self.do_convert_rgb}
def prepare_dummy_image(self):
img_url = "https://www.ilankelman.org/stopsigns/australia.jpg"
img_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/australia.jpg"
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
return raw_image
@ -87,7 +87,6 @@ class Pix2StructImageProcessingTest(ImageProcessingSavingTestMixin, unittest.Tes
self.assertTrue(hasattr(image_processor, "do_normalize"))
self.assertTrue(hasattr(image_processor, "do_convert_rgb"))
@unittest.skip("fix me Younes.")
def test_expected_patches(self):
dummy_image = self.image_processor_tester.prepare_dummy_image()