Skip to content

Accept custom rsl-rl log writers in the runner config. - #1257

Merged
kevinzakka merged 2 commits into
mainfrom
lll/feat/custom-log-writer
Nov 1, 2026
Merged

kevinzakka merged 2 commits into
mainfrom
lll/feat/custom-log-writer

Conversation

@louislelay

@louislelay louislelay commented Oct 21, 2026 •

Copy link
Copy Markdown
Collaborator

RslRlBaseRunnerCfg.logger only accepted "wandb" or "tensorboard", so projects couldn't plug in their own rsl-rl LogWriter even though rsl-rl already supports it. This lets logger take a writer import path like my_pkg.writers:MyLogWriter, which the runner passes to rsl-rl in its dict form (also silencing rsl-rl's deprecation warning for the plain "wandb" string). W&B tags are now only applied with the W&B logger.

`RslRlBaseRunnerCfg.logger` only accepted `"wandb"` or `"tensorboard"`, so projects couldn't plug in their own rsl-rl `LogWriter` even though rsl-rl already supports it. This lets `logger` take a writer import path like `my_pkg.writers:MyLogWriter`, which the runner passes to rsl-rl in its dict form (also silencing rsl-rl's deprecation warning for the plain `"wandb"` string). W&B tags are now only applied with the W&B logger.
@louislelay
louislelay added this pull request to stack #1259 October 21, 2026 14:53
@louislelay
louislelay requested a review from kevinzakka October 21, 2026 14:54
@louislelay
louislelay marked this pull request as ready for review October 21, 2026 14:54

@kevinzakka kevinzakka left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice, thanks Louis! Any reason not to type it as Literal["wandb", "tensorboard"] | dict and pass the dict straight to rsl-rl, so custom writers can take kwargs and the built-ins keep CLI validation?

@kevinzakka kevinzakka left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Never mind, a dict union breaks tyro's CLI parsing.

@kevinzakka
kevinzakka merged commit 2ad0103 into main Nov 1, 2026
18 checks passed
@kevinzakka
kevinzakka deleted the lll/feat/custom-log-writer branch November 1, 2026 03:35
@louislelay

Copy link
Copy Markdown
Collaborator Author

Thanks! Appreciate it :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants