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

Fix resume_download future warning (#31007)

* Fix resume_download future warning

* better like this

* Add regression test
parent acbfaf69
...@@ -648,7 +648,7 @@ class PretrainedConfig(PushToHubMixin): ...@@ -648,7 +648,7 @@ class PretrainedConfig(PushToHubMixin):
) -> Tuple[Dict[str, Any], Dict[str, Any]]: ) -> Tuple[Dict[str, Any], Dict[str, Any]]:
cache_dir = kwargs.pop("cache_dir", None) cache_dir = kwargs.pop("cache_dir", None)
force_download = kwargs.pop("force_download", False) force_download = kwargs.pop("force_download", False)
resume_download = kwargs.pop("resume_download", False) resume_download = kwargs.pop("resume_download", None)
proxies = kwargs.pop("proxies", None) proxies = kwargs.pop("proxies", None)
token = kwargs.pop("token", None) token = kwargs.pop("token", None)
local_files_only = kwargs.pop("local_files_only", False) local_files_only = kwargs.pop("local_files_only", False)
......
...@@ -20,6 +20,7 @@ import sys ...@@ -20,6 +20,7 @@ import sys
import tempfile import tempfile
import unittest import unittest
import unittest.mock as mock import unittest.mock as mock
import warnings
from pathlib import Path from pathlib import Path
from huggingface_hub import HfFolder, delete_repo from huggingface_hub import HfFolder, delete_repo
...@@ -306,3 +307,10 @@ class ConfigTestUtils(unittest.TestCase): ...@@ -306,3 +307,10 @@ class ConfigTestUtils(unittest.TestCase):
self.assertTrue(config._has_non_default_generation_parameters()) self.assertTrue(config._has_non_default_generation_parameters())
config = BertConfig(min_length=0) # `min_length = 0` is a default generation kwarg config = BertConfig(min_length=0) # `min_length = 0` is a default generation kwarg
self.assertFalse(config._has_non_default_generation_parameters()) self.assertFalse(config._has_non_default_generation_parameters())
def test_loading_config_do_not_raise_future_warnings(self):
"""Regression test for https://github.com/huggingface/transformers/issues/31002."""
# Loading config should not raise a FutureWarning. It was the case before.
with warnings.catch_warnings():
warnings.simplefilter("error")
PretrainedConfig.from_pretrained("bert-base-uncased")
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