fix unit test

kaiko-ai · Nov 28, 2024 · fc0c09d · fc0c09d
1 parent 55a5d20
commit fc0c09d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py b/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py
@@ -13,7 +13,9 @@
 PREDS_ONE = torch.randint(0, NUM_CLASSES_ONE, (NUM_BATCHES, BATCH_SIZE, 32, 32))
 TARGET_ONE = torch.randint(0, NUM_CLASSES_ONE, (NUM_BATCHES, BATCH_SIZE, 32, 32))
 EXPECTED_ONE = {
-    "GeneralizedDiceScore": torch.tensor(0.3482658863067627),
+    "DiceScore (micro)": torch.tensor(0.3482658863067627),
+    "DiceScore (macro)": torch.tensor(0.34805023670196533),
+    "DiceScore (weighted)": torch.tensor(0.3484232723712921),
     "MeanIoU": torch.tensor(0.2109210342168808),
 }
 """Test features."""