Merge branch 'master' into dependabot-pip-requirements-mypy-1.18.1

Author: Borda

.azure/gpu-unittests.yml

@@ -26,7 +26,7 @@ jobs:
       matrix:
         "PyTorch | 2.0 oldest":
           # Torch does not have build wheels with old Torch versions for newer CUDA
-          docker-image: "ubuntu22.04-cuda12.1.1-py3.9-torch2.0"
+          docker-image: "ubuntu22.04-cuda12.1.1-py3.10-torch2.0"
           torch-ver: "2.0"
         "PyTorch | 2.X stable":
           docker-image: "ubuntu24.04-cuda12.6.3-py3.12-torch2.8"

.devcontainer/Dockerfile

@@ -1,7 +1,7 @@
 # See here for image contents: https://github.com/microsoft/vscode-dev-containers/tree/v0.194.0/containers/python-3/.devcontainer/base.Dockerfile
 
-# [Choice] Python version: 3, 3.9, 3.8, 3.7, 3.6
-ARG VARIANT="3.9"
+# [Choice] Python version: 3, 3.10, 3.11, 3.12
+ARG VARIANT="3.10"
 FROM mcr.microsoft.com/vscode/devcontainers/python:${VARIANT}
 
 # [Choice] Node.js version: none, lts/*, 16, 14, 12, 10
@@ -10,21 +10,26 @@ RUN if [ "${NODE_VERSION}" != "none" ]; then \
       su vscode -c "umask 0002 && . /usr/local/share/nvm/nvm.sh && nvm install ${NODE_VERSION} 2>&1"; \
     fi
 
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
+ENV UV_SYSTEM_PYTHON=1
+
 COPY requirements/ /tmp/pip-tmp/requirements/
 RUN \
-    pip3 install awscli && \
+    uv pip install awscli && \
     aws s3 sync --no-sign-request s3://sphinx-packages/ dist/  && \
-    # trying to resolve pesq installation issue
-    pip3 install -q "numpy<1.24" && \
-    pip3 --disable-pip-version-check --no-cache-dir install \
+    # trying to resolve pesq installation issue (numpy<1.24 uses distutils, removed in Python 3.12)
+    if python3 -c "import sys; exit(0 if sys.version_info < (3, 12) else 1)"; then \
+        uv pip install -q "numpy<1.24"; \
+    fi && \
+    uv pip install --no-cache --index-strategy unsafe-best-match \
       -r /tmp/pip-tmp/requirements/_devel.txt \
       -r /tmp/pip-tmp/requirements/_docs.txt \
       --extra-index-url="https://download.pytorch.org/whl/cpu/" \
       --find-links="dist/" && \
     rm -rf /tmp/pip-tmp
 
 # [Optional] If your pip requirements rarely change, uncomment this section to add them to the image.
-# RUN pip3 --disable-pip-version-check --no-cache-dir install -r /tmp/pip-tmp/requirements/base.txt \
+# RUN uv pip install --no-cache -r /tmp/pip-tmp/requirements/base.txt \
 #    && rm -rf /tmp/pip-tmp
 
 # [Optional] Uncomment this section to install additional OS packages.

.devcontainer/devcontainer.json

@@ -2,14 +2,14 @@
 // https://github.com/microsoft/vscode-dev-containers/tree/v0.194.0/containers/python-3
 {
   "name": "PyTorch Lightning Metrics",
-  "image": "pytorchlightning/torchmetrics:devcontainer-py3.9",
+  "image": "pytorchlightning/torchmetrics:devcontainer-py3.10",
   // If you want to use a different Python version, uncomment the build object below
   // "build": {
   //     "dockerfile": "Dockerfile",
   //     "context": "..",
   //     "args": {
-  //         // Update 'VARIANT' to pick a Python version: 3, 3.6, 3.7, 3.8, 3.9
-  //         "VARIANT": "3.9",
+  //         // Update 'VARIANT' to pick a Python version: 3, 3.10, 3.11, 3.12
+  //         "VARIANT": "3.10",
   //         // Options
   //         "NODE_VERSION": "none"
   //     }

.github/workflows/ci-checks.yml

@@ -33,8 +33,8 @@ jobs:
       import-name: "torchmetrics"
       testing-matrix: |
         {
-          "os": ["ubuntu-22.04", "macos-13", "windows-2022"],
-          "python-version": ["3.9", "3.11"]
+          "os": ["ubuntu-22.04", "macOS-14", "windows-2022"],
+          "python-version": ["3.10", "3.11"]
         }
 
   check-md-links:

.github/workflows/ci-integrate.yml

@@ -26,18 +26,17 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: ["ubuntu-22.04", "macOS-13", "windows-2022"]
-        python-version: ["3.9", "3.11"]
+        os: ["ubuntu-22.04", "macOS-14", "windows-2022"]
+        python-version: ["3.10", "3.11"]
         requires: ["oldest", "latest"]
-        exclude:
-          - { python-version: "3.10", requires: "oldest" }
-          - { python-version: "3.10", os: "windows" } # todo: https://discuss.pytorch.org/t/numpy-is-not-available-error/146192
-        include:
-          - { python-version: "3.10", requires: "latest", os: "ubuntu-22.04" }
-          # - { python-version: "3.10", requires: "latest", os: "macOS-14" } # M1 machine # todo: crashing for MPS out of memory
+        # exclude:
+        #   - { python-version: "3.10", requires: "oldest" }
+        #   - { python-version: "3.10", os: "windows" } # todo: https://discuss.pytorch.org/t/numpy-is-not-available-error/146192
     env:
       FREEZE_REQUIREMENTS: ${{ ! (github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/heads/release/')) }}
       UV_TORCH_BACKEND: "cpu"
+      # TODO: Remove this - Enable running MPS tests on this platform
+      DISABLE_MPS: ${{ startsWith(matrix.os, 'macOS') && '1' || '0' }}
 
     # Timeout: https://stackoverflow.com/a/59076067/4521646
     # seems that MacOS jobs take much more than orger OS

.github/workflows/ci-tests.yml

@@ -48,11 +48,9 @@ jobs:
           - "2.8.0"
         include:
           # cover additional python and PT combinations
-          - { os: "ubuntu-22.04", python-version: "3.9", pytorch-version: "2.0.1", requires: "oldest" }
+          - { os: "ubuntu-22.04", python-version: "3.10", pytorch-version: "2.0.1", requires: "oldest" }
           - { os: "ubuntu-22.04", python-version: "3.12", pytorch-version: "2.7.1" }
           - { os: "ubuntu-22.04", python-version: "3.12", pytorch-version: "2.8.0" }
-          # standard mac machine, not the M1
-          - { os: "macOS-13", python-version: "3.10", pytorch-version: "2.0.1" }
           # using the ARM based M1 machine
           - { os: "macOS-14", python-version: "3.10", pytorch-version: "2.0.1" }
           - { os: "macOS-14", python-version: "3.12", pytorch-version: "2.8.0" }

.github/workflows/docker-build.yml

@@ -35,7 +35,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python: ["3.9", "3.10", "3.11"]
+        python: ["3.10", "3.11"]
     steps:
       - name: Checkout
         uses: actions/checkout@v5
@@ -67,7 +67,6 @@ jobs:
         include:
           # These are the base images for PL release docker images,
           #  so include at least all the combinations in release-dockers.yml.
-          - { python: "3.9", pytorch: "2.0.1", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.10", pytorch: "2.0.1", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.1.2", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.2.2", cuda: "12.1.1", ubuntu: "22.04" }

.github/workflows/docs-build.yml

@@ -18,8 +18,9 @@ defaults:
 
 env:
   FREEZE_REQUIREMENTS: "1"
-  TORCH_URL: "https://download.pytorch.org/whl/cpu/"
   PYPI_CACHE: "_ci-cache_PyPI"
+  TORCH_URL: "https://download.pytorch.org/whl/cpu/"
+  UV_TORCH_BACKEND: "cpu"
   PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION: "python"
   TOKENIZERS_PARALLELISM: false
   SPHINX_MOCK_REQUIREMENTS: 0

.lightning/workflows/uinttests.yml .lightning/workflows/unittests.yml.lightning/workflows/uinttests.yml renamed to .lightning/workflows/unittests.yml

@@ -9,7 +9,7 @@ machine: "L4_X_2"
 parametrize:
   matrix:
     image: # note that this is setting also all oldest requirements which is linked to Torch == 2.0
-      - "pytorchlightning/torchmetrics:ubuntu22.04-cuda12.1.1-py3.9-torch2.0"
+      - "pytorchlightning/torchmetrics:ubuntu22.04-cuda12.1.1-py3.10-torch2.0"
       - "pytorchlightning/torchmetrics:ubuntu24.04-cuda12.6.3-py3.12-torch2.8"
     # testing: ["doctest", "common", "distributed"] # todo
   include: []
@@ -21,6 +21,7 @@ env:
   TEST_DIRS: "unittests"
   FREEZE_REQUIREMENTS: 1
   SKIP_SLOW_DOCTEST: 1
+  LIGHTNING_CI: "1"
 
 run: |
   whereis nvidia
@@ -81,7 +82,7 @@ run: |
   # UnitTesting common
   #if [ "$testing" == "common" ]; then
   python -m pytest ${TEST_DIRS} -v \
-    -m "not DDP" --numprocesses=6 --dist=loadfile \
+    -m "not DDP" --numprocesses=3 --dist=loadfile \
     --cov=torchmetrics --timeout=360 --durations=100 \
     --reruns 3 --reruns-delay 1
   # UnitTesting DDP

CHANGELOG.md

@@ -18,7 +18,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Changed
 
-- Defaulting Dice score `average="macro"` ([#3042](https://github.com/Lightning-AI/torchmetrics/pull/3042))
+-
 
 
 ### Deprecated
@@ -38,6 +38,23 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ---
 
+## [1.9.0] - 2026-03-05
+
+### Changed
+
+- Defaulting Dice score `average="macro"` ([#3042](https://github.com/Lightning-AI/torchmetrics/pull/3042))
+- Dropped Python 3.9 support, set 3.10 as minimum ([#3330](https://github.com/Lightning-AI/torchmetrics/pull/3330))
+- Replaced `pkg_resources` with `packaging` ([#3329](https://github.com/Lightning-AI/torchmetrics/pull/3329))
+
+### Fixed
+
+- Fixed device mismatch in `Metric` base class ([#3316](https://github.com/Lightning-AI/torchmetrics/pull/3316))
+- Fixed n-d slicing deprecation warning ([#3319](https://github.com/Lightning-AI/torchmetrics/pull/3319))
+- Fixed tensor copy warning in `logAUC` ([#3295](https://github.com/Lightning-AI/torchmetrics/pull/3295))
+- Fixed memory issues when computing retrieval metrics with high-valued indices ([#3291](https://github.com/Lightning-AI/torchmetrics/pull/3291))
+- Fixed race condition in `_safe_divide` by creating tensor directly on device ([#3284](https://github.com/Lightning-AI/torchmetrics/pull/3284))
+
+
 ## [1.8.2] - 2025-09-03
 
 ### Fixed