fixed metric test

hzavadil98 · hzavadil98 · commit 46798d24533c · 2025-02-05T11:47:58.000+01:00
diff --git a/tests/test_metrics.py b/tests/test_metrics.py
@@ -94,6 +94,6 @@ def test_accuracy():
 
     accuracy_score = accuracy(y_true, y_pred)
 
-    assert accuracy_score.allclose(torch.tensor(0.8), atol=1e-5), (
+    assert (torch.abs(torch.tensor(accuracy_score - 0.8)) < 1e-5), (
         f"Accuracy Score: {accuracy_score.item()}"
     )

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,6 @@ def test_accuracy():`
`94`	`94`
`95`	`95`	`accuracy_score = accuracy(y_true, y_pred)`
`96`	`96`
`97`		`- assert accuracy_score.allclose(torch.tensor(0.8), atol=1e-5), (`
	`97`	`+ assert (torch.abs(torch.tensor(accuracy_score - 0.8)) < 1e-5), (`
`98`	`98`	`f"Accuracy Score: {accuracy_score.item()}"`
`99`	`99`	`)`