Unverified Commit 0ad770c3 authored by Lucain's avatar Lucain Committed by GitHub
Browse files

Fix @require_read_token in tests (#29367)

parent bb4f816a
......@@ -38,7 +38,6 @@ from typing import Callable, Dict, Iterable, Iterator, List, Optional, Union
from unittest import mock
from unittest.mock import patch
import huggingface_hub
import urllib3
from transformers import logging as transformers_logging
......@@ -466,11 +465,11 @@ def require_read_token(fn):
"""
A decorator that loads the HF token for tests that require to load gated models.
"""
token = os.getenv("HF_HUB_READ_TOKEN", None)
token = os.getenv("HF_HUB_READ_TOKEN")
@wraps(fn)
def _inner(*args, **kwargs):
with patch.object(huggingface_hub.utils._headers, "get_token", return_value=token):
with patch("huggingface_hub.utils._headers.get_token", return_value=token):
return fn(*args, **kwargs)
return _inner
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment