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