Fix integration between classwise wrapper and metric tracker

[SkafteNicki]

What does this PR do?

Fixes #2994

Before submitting

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

---

📚 Documentation preview 📚: https://torchmetrics--3004.org.readthedocs.build/en/3004/

[codecov]

Codecov Report

Attention: Patch coverage is 60.00000% with 2 lines in your changes missing coverage. Please review.

Project coverage is 36%. Comparing base (d332ae3) to head (cc91d35).
Report is 2 commits behind head on master.

❗ There is a different number of reports uploaded between BASE (d332ae3) and HEAD (cc91d35). Click for more details.

HEAD has 56 uploads less than BASE

Flag	BASE (d332ae3)	HEAD (cc91d35)
macOS	6	3
cpu	28	14
python3.12	6	3
torch2.6.0	2	1
python3.10	18	9
torch2.0.1	4	2
Windows	4	2
torch2.6.0+cpu	2	1
torch2.0.1+cpu	6	3
Linux	18	9
torch2.4.1+cpu	4	2
python3.11	2	1
torch2.3.1+cpu	2	1
torch2.5.0+cpu	4	2
torch2.1.2+cpu	2	1
python3.9	2	1
torch2.2.2+cpu	2	1

Additional details and impacted files

@@           Coverage Diff            @@
##           master   #3004     +/-   ##
========================================
- Coverage      69%     36%    -33%     
========================================
  Files         338     352     +14     
  Lines       19464   19647    +183     
========================================
- Hits        13486    7138   -6348     
- Misses       5978   12509   +6531     

🚀 New features to boost your workflow:

• ❄ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.