Skip to content

Commit 196a0a6

Browse files
fix(elasticsearch): apply ruff format to test file
1 parent 4e6674a commit 196a0a6

1 file changed

Lines changed: 4 additions & 12 deletions

File tree

integrations/elasticsearch/tests/test_inference_hybrid_retriever.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)