From 49a32e8296ac583d9dae8b24216423446c5f0bef Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Mon, 1 Dec 2025 12:18:36 +0000 Subject: [PATCH 1/2] build(deps): bump torchmetrics from 0.9.1 to 1.8.2 in /tests/e2e Bumps [torchmetrics](https://github.com/Lightning-AI/torchmetrics) from 0.9.1 to 1.8.2. - [Release notes](https://github.com/Lightning-AI/torchmetrics/releases) - [Changelog](https://github.com/Lightning-AI/torchmetrics/blob/v1.8.2/CHANGELOG.md) - [Commits](https://github.com/Lightning-AI/torchmetrics/compare/v0.9.1...v1.8.2) --- updated-dependencies: - dependency-name: torchmetrics dependency-version: 1.8.2 dependency-type: direct:production update-type: version-update:semver-major ... Signed-off-by: dependabot[bot] --- tests/e2e/mnist_pip_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt index 6afcba78..ec87256b 100644 --- a/tests/e2e/mnist_pip_requirements.txt +++ b/tests/e2e/mnist_pip_requirements.txt @@ -2,5 +2,5 @@ torch==2.7.1+cu118 torchvision==0.22.1+cu118 pytorch_lightning==1.9.5 -torchmetrics==0.9.1 +torchmetrics==1.8.2 minio From a8ad559286b904f7b498f8c0c0b2bb1c2932f206 Mon Sep 17 00:00:00 2001 From: kryanbeane Date: Mon, 1 Dec 2025 13:57:59 +0000 Subject: [PATCH 2/2] Fix Accuracy API usage for torch metrics v 1.x+ --- tests/e2e/mnist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/e2e/mnist.py b/tests/e2e/mnist.py index bc9101a8..176d0a2f 100644 --- a/tests/e2e/mnist.py +++ b/tests/e2e/mnist.py @@ -99,8 +99,8 @@ def __init__(self, data_dir=PATH_DATASETS, hidden_size=64, learning_rate=2e-4): nn.Linear(hidden_size, self.num_classes), ) - self.val_accuracy = Accuracy() - self.test_accuracy = Accuracy() + self.val_accuracy = Accuracy(task="multiclass", num_classes=self.num_classes) + self.test_accuracy = Accuracy(task="multiclass", num_classes=self.num_classes) def forward(self, x): x = self.model(x)