From 0db5f66ddae6afd62454c418771286fa64dbea20 Mon Sep 17 00:00:00 2001 From: Bruno Machado <72039033+brunomachado37@users.noreply.github.com> Date: Thu, 12 Mar 2026 00:54:08 +0100 Subject: [PATCH] Add option to disable tags on WandB (#1339) Signed-off-by: Steven Palma Co-authored-by: Steven Palma --- src/lerobot/configs/default.py | 1 + src/lerobot/rl/wandb_utils.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lerobot/configs/default.py b/src/lerobot/configs/default.py index dcb0cbd54..3fb0c6c4e 100644 --- a/src/lerobot/configs/default.py +++ b/src/lerobot/configs/default.py @@ -47,6 +47,7 @@ class WandBConfig: notes: str | None = None run_id: str | None = None mode: str | None = None # Allowed values: 'online', 'offline' 'disabled'. Defaults to 'online' + add_tags: bool = True # If True, save configuration as tags in the WandB run. @dataclass diff --git a/src/lerobot/rl/wandb_utils.py b/src/lerobot/rl/wandb_utils.py index ee30b75df..e3190b6ce 100644 --- a/src/lerobot/rl/wandb_utils.py +++ b/src/lerobot/rl/wandb_utils.py @@ -98,7 +98,7 @@ class WandBLogger: entity=self.cfg.entity, name=self.job_name, notes=self.cfg.notes, - tags=cfg_to_group(cfg, return_list=True, truncate_tags=True), + tags=cfg_to_group(cfg, return_list=True, truncate_tags=True) if self.cfg.add_tags else None, dir=self.log_dir, config=cfg.to_dict(), # TODO(rcadene): try set to True