Fix Accuracy API usage for torch metrics v 1.x+

kryanbeane · openshift-merge-bot[bot] · commit b8e12197f600 · 2025-12-01T14:22:37.000Z
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -99,8 +99,8 @@ def __init__(self, data_dir=PATH_DATASETS, hidden_size=64, learning_rate=2e-4):`
`99`	`99`	`nn.Linear(hidden_size, self.num_classes),`
`100`	`100`	`)`
`101`	`101`
`102`		`- self.val_accuracy = Accuracy()`
`103`		`- self.test_accuracy = Accuracy()`
	`102`	`+ self.val_accuracy = Accuracy(task="multiclass", num_classes=self.num_classes)`
	`103`	`+ self.test_accuracy = Accuracy(task="multiclass", num_classes=self.num_classes)`
`104`	`104`
`105`	`105`	`def forward(self, x):`
`106`	`106`	`x = self.model(x)`