## Summary

<!-- What does this PR change and why? -->

## Type of change

- [ ] Bug fix
- [ ] New feature / example
- [ ] Documentation
- [ ] Benchmark / reproduction
- [ ] Refactor / cleanup

## Checklist

- [ ] `ruff check .` passes
- [ ] `pytest -q` passes
- [ ] `python demo.py` still runs and PASSES
- [ ] Added/updated tests for any behavior change
- [ ] Updated `CHANGELOG.md` (Unreleased)
- [ ] No trained weights or proprietary data added

## Notes

<!-- Anything reviewers should know. -->