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:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user