diff --git a/tests/modules/text_embedding/test_embedding.py b/tests/modules/text_embedding/test_embedding.py index b4e294cd..e51ca06b 100644 --- a/tests/modules/text_embedding/test_embedding.py +++ b/tests/modules/text_embedding/test_embedding.py @@ -1018,7 +1018,7 @@ def test_encoding_order(loaded_model: EmbeddingModule, truncate_input_tokens): # test order by comparing value of individual embeddings in sequence for i, e in enumerate(separate_vectors): - assert np.allclose(e, combined_vectors[i]) + assert np.isclose(e, combined_vectors[i]) # test expected failure case by reordering shifted_separate_vectors = separate_vectors[1:] + [separate_vectors[0]]