Skip to content

Commit 0a7ea8b

Browse files
committed
[Fix] log WandB scalars using correct step argument
1 parent 41d702f commit 0a7ea8b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrl/record/loggers/wandb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,6 @@ def log_str(self, name: str, value: str, step: int | None = None) -> None:
234234
table = wandb.Table(columns=["text"], data=[[value]])
235235

236236
if step is not None:
237-
self.experiment.log({name: table, "trainer/step": step})
237+
self.experiment.log({name: value}, step=step)
238238
else:
239239
self.experiment.log({name: table})

0 commit comments

Comments
 (0)