@@ -49,9 +49,7 @@ class TestElasticsearchInferenceHybridRetriever:
4949 @patch ("haystack_integrations.document_stores.elasticsearch.document_store.Elasticsearch" )
5050 def test_to_dict (self , _mock_elasticsearch_client ) -> None :
5151 doc_store = ElasticsearchDocumentStore ()
52- retriever = ElasticsearchInferenceHybridRetriever (
53- document_store = doc_store , inference_id = ".elser_model_2"
54- )
52+ retriever = ElasticsearchInferenceHybridRetriever (document_store = doc_store , inference_id = ".elser_model_2" )
5553 assert retriever .to_dict () == self .serialised
5654
5755 @patch ("haystack_integrations.document_stores.elasticsearch.document_store.Elasticsearch" )
@@ -140,9 +138,7 @@ def test_run(self):
140138 mock_store ._bm25_retrieval .return_value = [Document (content = "BM25 result" )]
141139 mock_store ._sparse_vector_retrieval_inference .return_value = [Document (content = "Sparse result" )]
142140
143- retriever = ElasticsearchInferenceHybridRetriever (
144- document_store = mock_store , inference_id = ".elser_model_2"
145- )
141+ retriever = ElasticsearchInferenceHybridRetriever (document_store = mock_store , inference_id = ".elser_model_2" )
146142 result = retriever .run (query = "test query" )
147143
148144 assert len (result ["documents" ]) == 2
@@ -183,9 +179,7 @@ def test_run_with_runtime_params(self):
183179 mock_store ._bm25_retrieval .return_value = [Document (content = "BM25 result" )]
184180 mock_store ._sparse_vector_retrieval_inference .return_value = [Document (content = "Sparse result" )]
185181
186- retriever = ElasticsearchInferenceHybridRetriever (
187- document_store = mock_store , inference_id = ".elser_model_2"
188- )
182+ retriever = ElasticsearchInferenceHybridRetriever (document_store = mock_store , inference_id = ".elser_model_2" )
189183 retriever .run (
190184 query = "test query" ,
191185 filters_bm25 = {"field" : "value" },
@@ -213,9 +207,7 @@ def test_run_in_pipeline(self):
213207 mock_store ._bm25_retrieval .return_value = [Document (content = "BM25 result" )]
214208 mock_store ._sparse_vector_retrieval_inference .return_value = [Document (content = "Sparse result" )]
215209
216- retriever = ElasticsearchInferenceHybridRetriever (
217- document_store = mock_store , inference_id = ".elser_model_2"
218- )
210+ retriever = ElasticsearchInferenceHybridRetriever (document_store = mock_store , inference_id = ".elser_model_2" )
219211 pipeline = Pipeline ()
220212 pipeline .add_component ("retriever" , retriever )
221213
0 commit comments