diff --git a/tests/hub/test_patch_hf.py b/tests/hub/test_patch_hf.py deleted file mode 100644 index 13754923..00000000 --- a/tests/hub/test_patch_hf.py +++ /dev/null @@ -1,16 +0,0 @@ -import unittest - -from modelscope.msdatasets import MsDataset -from modelscope.utils.test_utils import test_level - - -class DownloadDatasetTest(unittest.TestCase): - - def setUp(self): - from modelscope.utils.hf_util import patch_hub - patch_hub() - - def test_automodel_download(self): - from transformers import AutoModel - model = AutoModel.from_pretrained('AI-ModelScope/bert-base-uncased') - self.assertTrue(model is not None)