Skip to content

Commit 2123476

Browse files
Block-hf-download-in-ci (#2696)
1 parent b638352 commit 2123476

File tree

5 files changed

+11
-5
lines changed

5 files changed

+11
-5
lines changed

tests/transformers/auto/test_configuration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from paddleformers.transformers.bert.configuration import BertConfig
2626
from paddleformers.utils.download import DownloadSource
2727
from paddleformers.utils.env import CONFIG_NAME
28-
from tests.testing_utils import set_proxy
28+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
2929

3030
from ...utils.test_module.custom_configuration import CustomConfig
3131

@@ -60,6 +60,7 @@ def test_community_model_class(self):
6060
auto_config = AutoConfig.from_pretrained(tempdir)
6161
self.assertEqual(auto_config.hidden_size, number)
6262

63+
@skip_for_none_ce_case
6364
@set_proxy(DownloadSource.HUGGINGFACE)
6465
def test_from_hf_hub(self):
6566
config = AutoConfig.from_pretrained("dfargveazd/tiny-random-llama-paddle-safe", download_hub="huggingface")

tests/transformers/auto/test_modeling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
from paddleformers.transformers.auto.modeling import MODEL_MAPPING
3636
from paddleformers.utils.download import DownloadSource
3737
from paddleformers.utils.env import CONFIG_NAME, PADDLE_WEIGHTS_NAME
38-
from tests.testing_utils import set_proxy
38+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
3939

4040
from ...utils.test_module.custom_configuration import CustomConfig
4141
from ...utils.test_module.custom_model import CustomModel
@@ -76,6 +76,7 @@ def test_model_from_pretrained_cache_dir(self):
7676
self.assertFalse(os.path.exists(os.path.join(tempdir, model_name, model_name)))
7777

7878
# @unittest.skip("skipping due to connection error!")
79+
@skip_for_none_ce_case
7980
@set_proxy(DownloadSource.HUGGINGFACE)
8081
def test_from_hf_hub(self):
8182
model = AutoModel.from_pretrained(

tests/transformers/test_configuration_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from paddleformers.utils import CONFIG_NAME
2828
from paddleformers.utils.download import DownloadSource
2929
from paddleformers.utils.env import LEGACY_CONFIG_NAME
30-
from tests.testing_utils import set_proxy
30+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
3131

3232

3333
class FakeSimplePretrainedModelConfig(PretrainedConfig):
@@ -154,6 +154,7 @@ def test_from_pretrained_cache_dir(self):
154154
# check against double appending model_name in cache_dir
155155
self.assertFalse(os.path.exists(os.path.join(tempdir, model_id, model_id)))
156156

157+
@skip_for_none_ce_case
157158
@set_proxy(DownloadSource.HUGGINGFACE)
158159
def test_load_from_hf(self):
159160
"""test load config from hf"""

tests/transformers/test_hf_tokenizer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
from paddleformers.transformers import AutoTokenizer, Qwen2Tokenizer
2020
from paddleformers.utils.download import DownloadSource
21-
from tests.testing_utils import set_proxy
21+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
2222

2323

2424
class TestHFMultiSourceTokenizer(unittest.TestCase):
@@ -42,6 +42,7 @@ def test_model_scope(self):
4242
tokenizer = Qwen2Tokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct", download_hub="modelscope")
4343
self.encode(tokenizer)
4444

45+
@skip_for_none_ce_case
4546
@set_proxy(DownloadSource.HUGGINGFACE)
4647
def test_hf_hub(self):
4748
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct", download_hub="huggingface")
@@ -55,6 +56,7 @@ def test_default(self):
5556
tokenizer = Qwen2Tokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
5657
self.encode(tokenizer)
5758

59+
@skip_for_none_ce_case
5860
@set_proxy(DownloadSource.HUGGINGFACE)
5961
def test_ernie_4_5_tokenizer(self):
6062
tokenizer = AutoTokenizer.from_pretrained("baidu/ERNIE-4.5-21B-A3B-PT", download_hub="huggingface")

tests/transformers/test_modeling_common.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
from paddleformers.transformers.model_utils import PretrainedModel
4040
from paddleformers.utils.env import CONFIG_NAME, LEGACY_CONFIG_NAME # MODEL_HOME,
4141

42-
from ..testing_utils import slow
42+
from ..testing_utils import skip_for_none_ce_case, slow
4343

4444

4545
def _config_zero_init(config):
@@ -785,6 +785,7 @@ class ModelTesterPretrainedMixin:
785785

786786
# Download from HF doesn't work in CI yet
787787
@slow
788+
@skip_for_none_ce_case
788789
def test_model_from_pretrained_hf_hub(self):
789790
if self.hf_remote_test_model_path is None or self.base_model_class is None:
790791
return

0 commit comments

Comments
 (0)