Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: fail to run ut in local because TEST_TOKEN required to set

Browse Source
This commit is contained in:
wenmeng.zwm
2023-02-22 14:21:06 +08:00
parent ac6979797b
commit cbea03a501
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modelscope/utils/test_utils.py
View File

@@ -22,6 +22,18 @@ from modelscope.utils.import_utils import is_tf_available, is_torch_available
TEST_LEVEL = 2
TEST_LEVEL_STR = 'TEST_LEVEL'
# for user citest and sdkdev
TEST_ACCESS_TOKEN1 = os.environ.get('TEST_ACCESS_TOKEN_CITEST', None)
TEST_ACCESS_TOKEN2 = os.environ.get('TEST_ACCESS_TOKEN_SDKDEV', None)
TEST_MODEL_CHINESE_NAME = '内部测试模型'
TEST_MODEL_ORG = 'citest'
def delete_credential():
path_credential = expanduser(DEFAULT_CREDENTIALS_PATH)
shutil.rmtree(path_credential, ignore_errors=True)
def test_level():
global TEST_LEVEL