Skip to content

unpin torchmetrics#38

Merged
sivanravidos merged 1 commit into
mainfrom
unpin_torechmetrics
Jun 28, 2026
Merged

unpin torchmetrics#38
sivanravidos merged 1 commit into
mainfrom
unpin_torechmetrics

Conversation

@sivanravidos

@sivanravidos sivanravidos commented Jun 28, 2026

Copy link
Copy Markdown
Collaborator

torchmetrics was pinned to 1.1 which prevents integration into algorithm-nexus since it collides with other packages.
This PR unpins its. The only effect I found is removed "pop" in MetricCollection which is bypassed by rebuilding the metric collection where needed.

@sivanravidos sivanravidos marked this pull request as ready for review June 28, 2026 08:19
@sivanravidos sivanravidos requested a review from mmdanziger June 28, 2026 08:19
@mmdanziger mmdanziger force-pushed the unpin_torechmetrics branch from 8c5f1e4 to 071c3ad Compare June 28, 2026 11:35
Removes dependency on MetricCollection.pop() which is absent in
torchmetrics>=1.2, unblocking the torchmetrics unpin. Callers pass
exclude={"perplexity"} when multiple tasks share a metric_key.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
@mmdanziger mmdanziger force-pushed the unpin_torechmetrics branch from 071c3ad to 54db755 Compare June 28, 2026 11:46

@mmdanziger mmdanziger left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM

@sivanravidos sivanravidos merged commit 902745f into main Jun 28, 2026
7 of 8 checks passed
@sivanravidos sivanravidos deleted the unpin_torechmetrics branch June 28, 2026 13:27
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