Skip to content

Commit f979acb

Browse files
authored
Fixed failing test_miou on M1 with pytest.approx (#3330)
1 parent 48c04de commit f979acb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/ignite/metrics/test_confusion_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def test_miou():
261261

262262
res = iou_metric.compute().numpy()
263263

264-
assert res == true_res_
264+
assert pytest.approx(res) == true_res_
265265

266266
for ignore_index in range(3):
267267
cm = ConfusionMatrix(num_classes=3)
@@ -271,7 +271,7 @@ def test_miou():
271271
cm.update(output)
272272
res = iou_metric.compute().numpy()
273273
true_res_ = np.mean(true_res[:ignore_index] + true_res[ignore_index + 1 :])
274-
assert res == true_res_, f"{ignore_index}: {res} vs {true_res_}"
274+
assert pytest.approx(res) == true_res_, f"{ignore_index}: {res} vs {true_res_}"
275275

276276

277277
def test_cm_accuracy():

0 commit comments

Comments
 (0)