From 36fbf2a6b74c2084957633ad8904119848505c81 Mon Sep 17 00:00:00 2001
From: Logic <logic@gitea.ideaopen.cn>
Date: Fri, 27 Mar 2026 14:52:22 +0800
Subject: [PATCH] fix: avoid swanlab run collisions for pusht configs

---
 image_pusht_diffusion_policy_dit.yaml     |  4 ++-
 image_pusht_diffusion_policy_dit_imf.yaml |  4 ++-
 tests/test_pusht_swanlab_config.py        | 32 +++++++++++++++++++++++
 3 files changed, 38 insertions(+), 2 deletions(-)
 create mode 100644 tests/test_pusht_swanlab_config.py

diff --git a/image_pusht_diffusion_policy_dit.yaml b/image_pusht_diffusion_policy_dit.yaml
index 1987aad..752a030 100644
--- a/image_pusht_diffusion_policy_dit.yaml
+++ b/image_pusht_diffusion_policy_dit.yaml
@@ -11,8 +11,10 @@ policy:
 logging:
   backend: swanlab
   mode: online
+  name: ${exp_name}
+  resume: false
   tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
-  id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
+  id: null
   group: ${exp_name}
 
 dataloader:
diff --git a/image_pusht_diffusion_policy_dit_imf.yaml b/image_pusht_diffusion_policy_dit_imf.yaml
index 804a80d..d4265f3 100644
--- a/image_pusht_diffusion_policy_dit_imf.yaml
+++ b/image_pusht_diffusion_policy_dit_imf.yaml
@@ -13,8 +13,10 @@ policy:
 logging:
   backend: swanlab
   mode: online
+  name: ${exp_name}
+  resume: false
   tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
-  id: ${now:%Y%m%d%H%M%S}_${name}_${task_name}
+  id: null
   group: ${exp_name}
 
 dataloader:
diff --git a/tests/test_pusht_swanlab_config.py b/tests/test_pusht_swanlab_config.py
new file mode 100644
index 0000000..130b11c
--- /dev/null
+++ b/tests/test_pusht_swanlab_config.py
@@ -0,0 +1,32 @@
+import pathlib
+
+from omegaconf import OmegaConf
+
+
+ROOT_DIR = pathlib.Path(__file__).resolve().parents[1]
+
+
+def _load_cfg(name: str):
+    return OmegaConf.load(ROOT_DIR / name)
+
+
+def test_image_pusht_dit_swanlab_config_uses_exp_name_and_no_resume_collision():
+    cfg = _load_cfg('image_pusht_diffusion_policy_dit.yaml')
+
+    assert cfg.logging.backend == 'swanlab'
+    assert cfg.logging.mode == 'online'
+    assert cfg.logging.name == cfg.exp_name
+    assert cfg.logging.resume is False
+    assert cfg.logging.id is None
+    assert cfg.logging.group == cfg.exp_name
+
+
+def test_image_pusht_dit_imf_swanlab_config_uses_exp_name_and_no_resume_collision():
+    cfg = _load_cfg('image_pusht_diffusion_policy_dit_imf.yaml')
+
+    assert cfg.logging.backend == 'swanlab'
+    assert cfg.logging.mode == 'online'
+    assert cfg.logging.name == cfg.exp_name
+    assert cfg.logging.resume is False
+    assert cfg.logging.id is None
+    assert cfg.logging.group == cfg.exp_name