Merge pull request #1288 from modelscope/fix/hf_util_ut

Add login for HFUtilTest
This commit is contained in:
Xingjun.Wang
2025-03-27 22:47:16 +08:00
committed by GitHub

View File

@@ -11,7 +11,8 @@ from huggingface_hub import CommitInfo, RepoUrl
from modelscope import HubApi
from modelscope.utils.hf_util.patcher import patch_context
from modelscope.utils.logger import get_logger
from modelscope.utils.test_utils import TEST_MODEL_ORG, test_level
from modelscope.utils.test_utils import (TEST_ACCESS_TOKEN1, TEST_MODEL_ORG,
test_level)
logger = get_logger()
@@ -21,6 +22,7 @@ class HFUtilTest(unittest.TestCase):
def setUp(self):
logger.info('SetUp')
self.api = HubApi()
self.api.login(TEST_ACCESS_TOKEN1)
self.user = TEST_MODEL_ORG
print(self.user)
self.create_model_name = '%s/%s_%s' % (self.user, 'test_model_upload',