Skip to content

feat: Add W&B run config API#615

Merged
vivekkalyan merged 1 commit intomainfrom
feat/wandb-config
Mar 12, 2026
Merged

feat: Add W&B run config API#615
vivekkalyan merged 1 commit intomainfrom
feat/wandb-config

Conversation

@vivekkalyan
Copy link
Collaborator

@vivekkalyan vivekkalyan commented Mar 12, 2026

Summary

  • add a public Model.update_wandb_config() API for attaching run config to ART-managed W&B runs
  • seed config at wandb.init(...) and sync it onto active runs without exposing mutable-config semantics
  • reject conflicting updates so run config stays immutable once set
  • add unit coverage for init-time config logging, active-run sync, and conflict rejection

Testing

  • uv run pytest tests/unit/test_metric_routing.py -q
  • uv run pytest tests/unit/test_frontend_logging.py -q
  • uv run ruff check src/art/model.py tests/unit/test_metric_routing.py
  • uv run ruff format --check src/art/model.py tests/unit/test_metric_routing.py

Downstream verification

  • pointed at this branch locally
  • logged a sanitized training config through ART and verified it appeared on the W&B run config

@vivekkalyan vivekkalyan requested a review from bradhilton March 12, 2026 19:17
@vivekkalyan vivekkalyan merged commit ca77e97 into main Mar 12, 2026
5 checks passed
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