partners: Use simsimd types (#25299)

The simsimd package [now has
types](https://github.com/ashvardanian/SimSIMD/releases/tag/v5.0.0)
This commit is contained in:
Christophe Bornet
2024-08-23 16:41:39 +02:00
committed by GitHub
parent 642f9530cd
commit 7f1e444efa
18 changed files with 1163 additions and 470 deletions

View File

@@ -77,7 +77,7 @@ def invoke_cosine_similarity_top_k_score_threshold(
expected_scores = [1.0, 0.93419873, 0.87038828, 0.83743579]
actual_idxs, actual_scores = cosine_similarity_top_k(X, Y, score_threshold=0.8)
assert actual_idxs == expected_idxs
assert np.allclose(expected_scores, actual_scores)
assert np.allclose(expected_scores, actual_scores, rtol=1.0e-3)
def test_cosine_similarity_top_k_and_score_threshold(