From 0ad770c3733f9478a8d9d0bc18cc6143877b47a2 Mon Sep 17 00:00:00 2001 From: Lucain Date: Thu, 29 Feb 2024 11:25:16 +0100 Subject: [PATCH] Fix @require_read_token in tests (#29367) --- src/transformers/testing_utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py index 0efaf77a98..fd5974d8d5 100644 --- a/src/transformers/testing_utils.py +++ b/src/transformers/testing_utils.py @@ -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