Fix elastic

deepset-ai · Jun 3, 2024 · 558a9b9 · 558a9b9
1 parent c0ce4d2
commit 558a9b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/integrations/elasticsearch/tests/test_bm25_retriever.py b/integrations/elasticsearch/tests/test_bm25_retriever.py
@@ -80,7 +80,6 @@ def test_run():
         fuzziness="AUTO",
         top_k=10,
         scale_score=False,
-        filter_policy="replace",
     )
     assert len(res) == 1
     assert len(res["documents"]) == 1

diff --git a/integrations/elasticsearch/tests/test_embedding_retriever.py b/integrations/elasticsearch/tests/test_embedding_retriever.py
@@ -37,6 +37,7 @@ def test_to_dict(_mock_elasticsearch_client):
             },
             "filters": {},
             "top_k": 10,
+            "filter_policy": "replace",
             "num_candidates": None,
         },
     }
@@ -54,6 +55,7 @@ def test_from_dict(_mock_elasticsearch_client):
             },
             "filters": {},
             "top_k": 10,
+            "filter_policy": "replace",
             "num_candidates": None,
         },
     }