From 00df1233095a2ccb31e328f6d7515643dcf58efb Mon Sep 17 00:00:00 2001 From: taprosoft Date: Fri, 27 Sep 2024 04:18:11 +0000 Subject: [PATCH] fix: fix vectorstore tests #none --- libs/kotaemon/tests/test_vectorstore.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/kotaemon/tests/test_vectorstore.py b/libs/kotaemon/tests/test_vectorstore.py index 2bf22ed..d148e92 100644 --- a/libs/kotaemon/tests/test_vectorstore.py +++ b/libs/kotaemon/tests/test_vectorstore.py @@ -65,7 +65,7 @@ class TestChromaVectorStore: db.add(embeddings=embeddings, metadatas=metadatas, ids=ids) _, sim, out_ids = db.query(embedding=[0.1, 0.2, 0.3], top_k=1) - assert sim == [1.0] + assert sim[0] - 1.0 < 1e-6 assert out_ids == ["a"] _, _, out_ids = db.query(embedding=[0.42, 0.52, 0.53], top_k=1) @@ -224,7 +224,7 @@ class TestMilvusVectorStore: db.add(embeddings=normalized_embeddings, metadatas=metadatas, ids=ids) _, sim, out_ids = db.query(embedding=normalized_embeddings[0], top_k=1) - assert sim == [1.0] + assert sim[0] - 1.0 < 1e-6 assert out_ids == ["a"] query_embedding = [