Merge pull request #9 from flaviabeo/tolerance

Adds tolerance params
caikit · Jul 17, 2024 · 0cf2c94 · 0cf2c94
2 parents 59ae6d5 + 3cecf52
commit 0cf2c94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modules/text_embedding/test_embedding.py b/tests/modules/text_embedding/test_embedding.py
@@ -1029,7 +1029,7 @@ def test_encoding_order(loaded_model: EmbeddingModule, truncate_input_tokens):
             not approx(e) == combined_vectors[i]
         ), "expected altered order to not match combined vectors"
         assert not np.allclose(
-            e, combined_vectors[i]
+            e, combined_vectors[i], rtol=1e-05, atol=1e-08
         ), "expected altered order to not match combined"