From c5be16950f5eebe52ca0ab8a2ec9506d12a9a70b Mon Sep 17 00:00:00 2001
From: "wenmeng.zwm" <wenmeng.zwm@alibaba-inc.com>
Date: Wed, 22 Feb 2023 14:21:06 +0800
Subject: [PATCH] fix: fail to run ut in local because TEST_TOKEN required to
 set

---
 modelscope/utils/test_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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'