adjust tolerance

datastax · Sep 26, 2024 · 9dd40d2 · 9dd40d2
1 parent 7a36a93
commit 9dd40d2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/libs/knowledge-store/tests/unit_tests/test_mmr_helper.py b/libs/knowledge-store/tests/unit_tests/test_mmr_helper.py
@@ -74,7 +74,7 @@ def test_mmr_helper_added_documetns() -> None:
     )
     assert helper.pop_best() == "v2"
 
-    assert math.isclose(helper.selected_similarity_scores[0], 0.9250772068)
-    assert math.isclose(helper.selected_similarity_scores[1], 0.7071067811)
-    assert math.isclose(helper.selected_mmr_scores[0], 0.4625386034)
-    assert math.isclose(helper.selected_mmr_scores[1], 0.1607613986)
+    assert math.isclose(helper.selected_similarity_scores[0], 0.9251, abs_tol=0.0001)
+    assert math.isclose(helper.selected_similarity_scores[1], 0.7071, abs_tol=0.0001)
+    assert math.isclose(helper.selected_mmr_scores[0], 0.4625, abs_tol=0.0001)
+    assert math.isclose(helper.selected_mmr_scores[1], 0.1608, abs_tol=0.0001)