Add unit tests for SwanlabMonitor functionality

Author: binary-husky

tests/utils/swanlab_test.py

@@ -0,0 +1,40 @@
+import os
+import unittest
+
+
+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"
+
+    @classmethod
+    def tearDownClass(cls):
+        # Restore original environment variables
+        for k, v in cls._original_env.items():
+            if v is None:
+                os.environ.pop(k, None)
+            else:
+                os.environ[k] = v
+
+    def test_swanlab_monitor_smoke(self):
+        from trinity.utils.monitor import SwanlabMonitor
+
+        # Try creating the monitor; if swanlab isn't installed, __init__ will assert
+        mon = SwanlabMonitor(
+            project="trinity-smoke",
+            group="cradle",
+            name="swanlab-env",
+            role="tester",
+        )
+
+        # Log a minimal metric to verify basic flow
+        mon.log({"smoke/metric": 1.0}, step=1)
+        mon.close()
+
+
+if __name__ == "__main__":
+    unittest.main()

trinity/utils/monitor.py

@@ -308,7 +308,7 @@ def __init__(
             "mode": monitor_args.get("mode") or "cloud",
             "settings": monitor_args.get("settings"),
             "id": monitor_args.get("id"),
-            "config": config.flatten(),
+            "config": config.flatten() if config is not None else None,
             "resume": monitor_args.get("resume"),
             "reinit": monitor_args.get("reinit"),
         }