fix: test error in cloud mode

Author: SAKURA-CAT

swanlab/data/run/main.py

@@ -86,7 +86,8 @@ def __init__(
             should_save=not self.__operator.disabled,
             version=get_package_version(),
         )
-        self.__public = SwanLabPublicConfig(self.__project_name, self.__settings)
+        self.__mode = get_mode()
+        self.__public = SwanLabPublicConfig(self.__project_name, self.__settings, self.__mode)
         self.__operator.before_run(self.__settings)
         # ---------------------------------- 初始化日志记录器 ----------------------------------
         swanlog.level = check_log_level(log_level)
@@ -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

@@ -4,13 +4,15 @@
 from swanlab.package import get_project_url, get_experiment_url
 
 
-class SwanlabCloudConfig:
+class SwanlabModeConfig:
     """
-    public data for the SwanLab project when running in cloud mode.
+    public data for the SwanLab project when running in different modes.
+    Only available in cloud mode.
     """
 
-    def __init__(self):
+    def __init__(self, mode: str):
         self.__http = None
+        self.__mode = mode
 
     def __get_property_from_http(self, name: str):
         """
@@ -27,6 +29,8 @@ def available(self):
         """
         Whether the SwanLab is running in cloud mode.
         """
+        if self.__mode != "cloud":
+            return False
         try:
             if self.__http is None:
                 self.__http = get_http()
@@ -79,9 +83,9 @@ class SwanLabPublicConfig:
     Public data for the SwanLab project.
     """
 
-    def __init__(self, project_name: str, settings: SwanLabSharedSettings):
+    def __init__(self, project_name: str, settings: SwanLabSharedSettings, mode: str):
         self.__project_name = project_name
-        self.__cloud = SwanlabCloudConfig()
+        self.__cloud = SwanlabModeConfig(mode)
         self.__settings = settings
 
     def json(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):