Skip to content

Commit 2321c5d

Browse files
committed
update diff
1 parent b588f80 commit 2321c5d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/accuracy/test_accuracy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,12 +158,12 @@ def compare_classification_result(outputs: ClassificationResult, reference: dict
158158
for i, (actual_label, expected_label) in enumerate(zip(outputs.top_labels, reference["top_labels"])):
159159
assert actual_label.id == expected_label["id"], f"Label {i} id mismatch"
160160
assert actual_label.name == expected_label["name"], f"Label {i} name mismatch"
161-
assert abs(actual_label.confidence - expected_label["confidence"]) < 1e-2, f"Label {i} confidence mismatch"
161+
assert abs(actual_label.confidence - expected_label["confidence"]) < 1e-1, f"Label {i} confidence mismatch"
162162

163163
assert "raw_scores" in reference
164164
assert outputs.raw_scores is not None
165165
expected_scores = np.array(reference["raw_scores"])
166-
assert np.allclose(outputs.raw_scores, expected_scores, rtol=1e-5, atol=1e-5), "raw_scores mismatch"
166+
assert np.allclose(outputs.raw_scores, expected_scores, rtol=1e-2, atol=1e-1), "raw_scores mismatch"
167167

168168

169169
def create_classification_result_dump(outputs: ClassificationResult) -> dict:

0 commit comments

Comments
 (0)