NVIDIA-Merlin / NVIDIA-Merlin/Transformers4Rec

[FEA] unpin `torchmetrics`

Open
#489 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

status/needs-triage
Dominant language
Python
Stars
1.3k
Forks
165
Avg merge
1m
Merged PRs (30d)
2

Description

🚀 Feature request

Would it be possible to unpin torchmetics and use some newer version, e.g. >=0.7?

Motivation

Seems there are some limitations/constraints on using other latest packages, see for example https://github.com/Lightning-AI/lightning/issues/14729

Your contribution

I think we can send a PR with this change.
Also, we could offer to use Ecosystem CI to gain some visibility on future compatibility
see Stay Ahead of Breaking Changes with the New Lightning Ecosystem CI

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by locating the dependency declaration that pins torchmetrics and review the project's compatibility checks. Update the allowed version to include newer releases, then run the relevant test suite and confirm the package installs and remains compatible with the supported PyTorch environment.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.