From c45f9b99b40ac229b457e169d367a0da8eab4e85 Mon Sep 17 00:00:00 2001 From: ilai Date: Thu, 28 Sep 2023 10:58:19 +0300 Subject: [PATCH] [test] Bug fix after defaults change --- tests/system/knowledge_base/test_knowledge_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/system/knowledge_base/test_knowledge_base.py b/tests/system/knowledge_base/test_knowledge_base.py index 0d2ffd9c..836bb56c 100644 --- a/tests/system/knowledge_base/test_knowledge_base.py +++ b/tests/system/knowledge_base/test_knowledge_base.py @@ -168,7 +168,7 @@ def test_query(knowledge_base, encoded_chunks): assert len(query_results) == 2 - expected_top_k = [10, 2] + expected_top_k = [5, 2] expected_first_results = [DocumentWithScore(id=chunk.id, text=chunk.text, metadata=chunk.metadata,