Skip to content

feat(plugin): add inspect-robots-wandb plugin for streaming metrics to Weights & Biases - #366

Open
ujjwalredd wants to merge 1 commit into
robocurve:mainfrom
ujjwalredd:feat/plugin-wandb-sink
Open

feat(plugin): add inspect-robots-wandb plugin for streaming metrics to Weights & Biases#366
ujjwalredd wants to merge 1 commit into
robocurve:mainfrom
ujjwalredd:feat/plugin-wandb-sink

Conversation

@ujjwalredd

Copy link
Copy Markdown

Summary

Adds plugins/inspect-robots-wandb implementing WandbSink to stream evaluation runs, trial metrics, step counts, duration, and success rates to Weights & Biases dashboards.

Changes

  • Created plugins/inspect-robots-wandb/ package with pyproject.toml, README.md, and WandbSink implementation.
  • Registered entry point wandb = "inspect_robots_wandb.sink:wandb_sink" under inspect_robots.sinks.
  • Added unit tests in plugins/inspect-robots-wandb/tests/test_wandb_sink.py achieving 100% test coverage.
  • Updated .github/workflows/ci.yml with plugin-wandb job and ci-ok dependency.
  • Updated CHANGELOG.md.

Checklist

  • pre-commit install done; hooks pass
  • Tests added/updated
  • Coverage stays at 100% (pytest --cov)
  • ruff check . and ruff format --check . pass
  • mypy passes (strict)
  • CHANGELOG.md updated under "Unreleased"
  • Public API changes are reflected in inspect_robots.__all__ and the API-snapshot test
  • Core stays NumPy-only (new deps are optional extras, lazily imported)

Related

Closes #365

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.

[feature] Add Weights & Biases (W&B) logging sink plugin (inspect-robots-wandb)

1 participant