diff --git a/modelscope/utils/test_utils.py b/modelscope/utils/test_utils.py index d3ed9978..291fa768 100644 --- a/modelscope/utils/test_utils.py +++ b/modelscope/utils/test_utils.py @@ -25,8 +25,8 @@ TEST_LEVEL = 2 TEST_LEVEL_STR = 'TEST_LEVEL' # for user citest and sdkdev -TEST_ACCESS_TOKEN1 = os.environ['TEST_ACCESS_TOKEN_CITEST'] -TEST_ACCESS_TOKEN2 = os.environ['TEST_ACCESS_TOKEN_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'