Bug

## 🐛 Bug <!-- A clear and concise description of what the bug is. --> ### To Reproduce Sample code and steps to reproduce the behavior with expected result... <!-- Create a new Lightning Studio with code that reproduces the issue and share the link. Also include all the relevant files and data required to reproduce shared issue. In case the code does not crash, please add assert statements to show what is the real and expected output. A simple guide on how to create such a studio can be found [here]( https://www.youtube.com/watch?v=YcW-2Zt_bFg&ab_channel=LightningAI). --> <details> <summary>Code sample</summary> ```python # Ideally attach a minimal code sample to reproduce the decried issue. # Minimal means having the shortest code but still preserving the bug. ``` </details> <details> <summary>Environment</summary> - TorchMetrics version (if build from source, add commit SHA): ??? - Python & PyTorch Version (e.g., 1.0): ??? - Any other relevant information such as OS (e.g., Linux): ??? </details> ### Additional context <!-- Add any other context about the problem here. -->