SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix test_load_img_url_timeout (#25976)

Browse Source 
* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-09-05 11:34:28 +02:00
committed by GitHub
parent feec56959a
commit fbbe1b8a40
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utils/test_image_utils.py
View File

@@ -21,7 +21,7 @@ import datasets
import numpy as np import numpy as np
import pytest import pytest
from huggingface_hub.file_download import http_get from huggingface_hub.file_download import http_get
from requests import ReadTimeout from requests import ConnectTimeout, ReadTimeout
from tests.pipelines.test_pipelines_document_question_answering import INVOICE_URL from tests.pipelines.test_pipelines_document_question_answering import INVOICE_URL
from transformers import is_torch_available, is_vision_available from transformers import is_torch_available, is_vision_available
@@ -491,7 +491,7 @@ class LoadImageTester(unittest.TestCase):
@is_flaky() @is_flaky()
def test_load_img_url_timeout(self): def test_load_img_url_timeout(self):
with self.assertRaises(ReadTimeout): with self.assertRaises((ReadTimeout, ConnectTimeout)):
load_image(INVOICE_URL, timeout=0.001) load_image(INVOICE_URL, timeout=0.001)
def test_load_img_local(self): def test_load_img_local(self):