Skip to content

Commit 5ebbc73

Browse files
committed
revert logger.py
Signed-off-by: Yuki Huang <[email protected]>
1 parent 14be342 commit 5ebbc73

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

nemo_rl/utils/logger.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -832,7 +832,7 @@ def __init__(self, cfg: LoggerConfig):
832832
self.wandb_logger = WandbLogger(cfg["wandb"], log_dir=wandb_log_dir)
833833
self.loggers.append(self.wandb_logger)
834834

835-
if cfg.get("swanlab_enabled", False):
835+
if cfg["swanlab_enabled"]:
836836
swanlab_log_dir = os.path.join(self.base_log_dir, "swanlab")
837837
os.makedirs(swanlab_log_dir, exist_ok=True)
838838
self.swanlab_logger = SwanlabLogger(cfg["swanlab"], log_dir=swanlab_log_dir)
@@ -862,7 +862,7 @@ def __init__(self, cfg: LoggerConfig):
862862
f"{metric_prefix}/*", step_metric=step_metric
863863
)
864864

865-
if cfg.get("swanlab_enabled", False) and self.swanlab_logger:
865+
if cfg["swanlab_enabled"] and self.swanlab_logger:
866866
self.swanlab_logger.define_metric(
867867
f"{metric_prefix}/*", step_metric=step_metric
868868
)

0 commit comments

Comments
 (0)