Update atol on precision at k test.

tensorflow · Jan 21, 2022 · a476c42 · a476c42
1 parent 9207f48
commit a476c42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/retrieval_metrics/test_precision_at_k.py b/tests/retrieval_metrics/test_precision_at_k.py
@@ -31,4 +31,4 @@ def test_compute(avg, expected):
     rm = PrecisionAtK(k=3, average=avg)
 
     precision = rm.compute(query_labels=query_labels, match_mask=match_mask)
-    np.testing.assert_allclose(precision, expected)
+    np.testing.assert_allclose(precision, expected, atol=1e-05)