Prepare tests for hfh 0.14 (#22958)

* Test hf_hub 0.14.0rc1

* fix mocked tests

* package version

---------

Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>
Co-authored-by: testbot <lucainp@hf.co>
This commit is contained in:
Lucain
2023-04-24 15:31:50 +02:00
committed by GitHub
parent 69f2d5386b
commit 74c55ab9e5
8 changed files with 17 additions and 12 deletions

View File

@@ -34,6 +34,7 @@ from typing import Iterator, List, Optional, Union
from unittest import mock
import huggingface_hub
import requests
from transformers import logging as transformers_logging
@@ -1688,12 +1689,16 @@ class RequestCounter:
self.head_request_count = 0
self.get_request_count = 0
self.other_request_count = 0
self.old_request = huggingface_hub.file_download.requests.request
huggingface_hub.file_download.requests.request = self.new_request
# Mock `get_session` to count HTTP calls.
self.old_get_session = huggingface_hub.utils._http.get_session
self.session = requests.Session()
self.session.request = self.new_request
huggingface_hub.utils._http.get_session = lambda: self.session
return self
def __exit__(self, *args, **kwargs):
huggingface_hub.file_download.requests.request = self.old_request
huggingface_hub.utils._http.get_session = self.old_get_session
def new_request(self, method, **kwargs):
if method == "GET":
@@ -1703,7 +1708,7 @@ class RequestCounter:
else:
self.other_request_count += 1
return self.old_request(method=method, **kwargs)
return requests.request(method=method, **kwargs)
def is_flaky(max_attempts: int = 5, wait_before_retry: Optional[float] = None, description: Optional[str] = None):