fix: test error in cloud mode

Author: SAKURA-CAT

swanlab/data/run/main.py

@@ -86,6 +86,7 @@ def __init__(
             should_save=not self.__operator.disabled,
             version=get_package_version(),
         )
+        self.__mode = get_mode()
         self.__public = SwanLabPublicConfig(self.__project_name, self.__settings)
         self.__operator.before_run(self.__settings)
         # ---------------------------------- 初始化日志记录器 ----------------------------------
@@ -216,7 +217,7 @@ def public(self):
 
     @property
     def mode(self) -> str:
-        return get_mode()
+        return self.__mode
 
     @property
     def state(self) -> SwanLabRunState:

swanlab/data/run/public.py

@@ -1,6 +1,7 @@
 from swankit.core import SwanLabSharedSettings
 
 from swanlab.api import get_http
+from swanlab.env import get_mode
 from swanlab.package import get_project_url, get_experiment_url
 
 
@@ -10,7 +11,13 @@ class SwanlabCloudConfig:
     """
 
     def __init__(self):
-        self.__http = None
+        # 不可以直接使用get_http，因为init时http对象还没有初始化
+        self.__mode = get_mode()
+        if self.__mode == "cloud":
+            self.__http = get_http()
+        else:
+            self.__http = None
+        self.__available = self.__http is not None
 
     def __get_property_from_http(self, name: str):
         """
@@ -27,12 +34,7 @@ def available(self):
         """
         Whether the SwanLab is running in cloud mode.
         """
-        try:
-            if self.__http is None:
-                self.__http = get_http()
-            return True
-        except ValueError:
-            return False
+        return self.__available
 
     @property
     def project_name(self):

test/unit/data/test_sdk.py

@@ -26,6 +26,9 @@ def setup_function():
     在当前测试文件下的每个测试函数执行前后执行
     """
     swanlog.disable_log()
+    run = get_run()
+    if run is not None:
+        run.finish()
     yield
     run = get_run()
     if run is not None:
@@ -153,6 +156,8 @@ def test_init_local(self):
         run = S.init(mode="local")
         assert os.environ[MODE] == "local"
         run.log({"TestInitMode": 1})  # 不会报错
+        assert run.mode == "local"
+        assert run.public.cloud.available is False
         assert get_run() is not None
         assert run.public.cloud.project_name is None
 
@@ -181,6 +186,17 @@ def test_init_error(self):
             S.init(mode="123456")  # noqa
         assert get_run() is None
 
+    @pytest.mark.skipif(T.is_skip_cloud_test, reason="skip cloud test")
+    def test_init_multiple(self):
+        # 先初始化cloud
+        self.test_init_cloud()
+        get_run().finish()
+        # 再初始化local
+        self.test_init_local()
+        get_run().finish()
+        # 再初始化disabled
+        self.test_init_disabled()
+
     # ---------------------------------- 测试环境变量输入 ----------------------------------
 
     def test_init_disabled_env(self):