fixed metric test

SFI-Visual-Intelligence · Feb 5, 2025 · 46798d2 · 46798d2
1 parent 0ebaced
commit 46798d2
Showing 1 changed file with 1 addition and 1 deletion.
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()}"
     )