Skip to content

refactor(aggregation): Add gramian_weighting field#533

Merged
ValerianRey merged 5 commits intomainfrom
add-gramian-weighting-field
Jan 28, 2026
Merged

refactor(aggregation): Add gramian_weighting field#533
ValerianRey merged 5 commits intomainfrom
add-gramian-weighting-field

Conversation

@ValerianRey
Copy link
Copy Markdown
Contributor

@ValerianRey ValerianRey commented Jan 28, 2026

  • refactor(aggregation): Add gramian_weighting field to GramianWeightedAggregator
  • Rename weighting to gramian_weighting in GramianWeightedAggregator
  • Use gramian_weighting in monitoring example

This fixes some typing problems and it's also much cleaner for users that want to hook the gramian weighting. I deliberately avoided calling it psd_weighting because it's somewhat user-facing (and we currently never really explain what psd is).

It's technically not a breaking change because aggregator.weighting.weighting still exists. So I think we don't have to add a changelog entry for this.

@ValerianRey ValerianRey added package: aggregation cc: refactor Conventional commit type for any refactoring, not user-facing, and not typing or perf improvements labels Jan 28, 2026
@codecov
Copy link
Copy Markdown

codecov bot commented Jan 28, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

Files with missing lines Coverage Δ
src/torchjd/aggregation/_aggregator_bases.py 100.00% <100.00%> (ø)
🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@ValerianRey ValerianRey marked this pull request as ready for review January 28, 2026 12:57
@ValerianRey ValerianRey changed the title refactor(aggregation): Add gramian_weighting field refactor(aggregation): Add gramian_weighting field Jan 28, 2026
Copy link
Copy Markdown
Contributor

@PierreQuinton PierreQuinton left a comment

Choose a reason for hiding this comment

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

LGTM. Arguably, adding a changlog entry would notify some users that they can do that now (but I don't really care either way is fine).

@ValerianRey ValerianRey merged commit 81648e5 into main Jan 28, 2026
16 checks passed
@ValerianRey ValerianRey deleted the add-gramian-weighting-field branch January 28, 2026 14:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cc: refactor Conventional commit type for any refactoring, not user-facing, and not typing or perf improvements package: aggregation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants