Refactor monitor module for improved structure and readability

Author: binary-husky

tests/utils/swanlab_test.py

@@ -5,11 +5,7 @@
 class TestSwanlabMonitor(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        # Ensure an env-based key path is exercised (uses dummy if not provided)
-        cls.env_keys = ["SWANLAB_API_KEY", "SWANLAB_APIKEY", "SWANLAB_KEY", "SWANLAB_TOKEN"]
-        cls._original_env = {k: os.environ.get(k) for k in cls.env_keys}
-        if not any(os.getenv(k) for k in cls.env_keys):
-            os.environ["SWANLAB_API_KEY"] = "dummy_key_for_smoke_test"
+        os.environ["SWANLAB_API_KEY"] = "xxxxxxxxxxxxxxxxxxxxx"
 
     @classmethod
     def tearDownClass(cls):

trinity/utils/monitor.py

@@ -34,6 +34,7 @@
         "tensorboard": "trinity.utils.monitor.TensorboardMonitor",
         "wandb": "trinity.utils.monitor.WandbMonitor",
         "mlflow": "trinity.utils.monitor.MlflowMonitor",
+        "swanlab": "trinity.utils.monitor.SwanlabMonitor",
     },
 )
 
@@ -240,7 +241,6 @@ def default_args(cls) -> Dict:
         }
 
 
-@MONITOR.register_module("swanlab")
 class SwanlabMonitor(Monitor):
     """Monitor with SwanLab.