feat: support swanlab logger

[terrykong]

Replacement of #716 (just resolving issues in the PR to help it along)

Thanks to @tpoisonooo for driving

[terrykong]

the doctest CI test failing is odd. still working on reproducing it locally

[terrykong]

ok, was able to pull the image that the CI used, for some reason the image is not correct. need to follow up with someone from @NVIDIA-NeMo/automation

[terrykong]

Thanks to @chtruong814 for helping solve this in #942

Retrying the CI. @yfw could you re-approve?

[github-actions]

❌ Submodule Fast-Forward Check Failed

Check based on commit: cdd5837 (PR #923 from tk/pr-716)

❌ Submodules that need attention:

Megatron-LM: ❌ PR branch is BEHIND main branch
TARGET (main branch): https://github.com/terrykong/Megatron-LM/commits/e2d5bcd605108e2cf64fdb91fdfc669f10a57f56/
CURRENT (PR #923 from tk/pr-716): https://github.com/terrykong/Megatron-LM/commits/2ff0f099ffc30ffd152e3e29e921a1609d00855c/

Please ensure all submodule commits are fast-forwards of the main branch before merging.

[github-actions]

❌ Submodule Fast-Forward Check Failed

Check based on commit: 397e420 (PR #923 from tk/pr-716)

❌ Submodules that need attention:

Megatron-LM: ❌ PR branch is BEHIND main branch
TARGET (main branch): https://github.com/terrykong/Megatron-LM/commits/e2d5bcd605108e2cf64fdb91fdfc669f10a57f56/
CURRENT (PR #923 from tk/pr-716): https://github.com/terrykong/Megatron-LM/commits/2ff0f099ffc30ffd152e3e29e921a1609d00855c/

Please ensure all submodule commits are fast-forwards of the main branch before merging.

[coderabbitai]

Warning

Rate limit exceeded

@terrykong has exceeded the limit for the number of commits or files that can be reviewed per hour. Please wait 6 minutes and 9 seconds before requesting another review.

⌛ How to resolve this issue?

After the wait time has elapsed, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

We recommend that you space out your commits to avoid hitting the rate limit.

🚦 How do rate limits work?

CodeRabbit enforces hourly rate limits for each developer per organization.

Our paid plans have higher rate limits than the trial, open-source and free plans. In all cases, we re-allow further reviews after a brief timeout.

Please see our FAQ for further information.

📥 Commits

Reviewing files that changed from the base of the PR and between e60c4d9 and 0f7fd8f.

⛔ Files ignored due to path filters (1)

• uv.lock is excluded by !**/*.lock

📒 Files selected for processing (12)

• docs/design-docs/logger.md (9 hunks)
• examples/configs/dpo.yaml (1 hunks)
• examples/configs/grpo_math_1B.yaml (1 hunks)
• examples/configs/grpo_math_1B_megatron.yaml (1 hunks)
• examples/configs/grpo_sliding_puzzle.yaml (1 hunks)
• examples/configs/rm.yaml (1 hunks)
• examples/configs/sft.yaml (1 hunks)
• examples/configs/sft_openmathinstruct2.yaml (1 hunks)
• examples/configs/vlm_grpo_3B.yaml (1 hunks)
• nemo_rl/utils/logger.py (7 hunks)
• pyproject.toml (1 hunks)
• tests/unit/utils/test_logger.py (22 hunks)

✨ Finishing touches

• 📝 Generate Docstrings

🧪 Generate unit tests

• Create PR with unit tests
• Post copyable unit tests in a comment
• Commit unit tests in branch tk/pr-716

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}