From 39e5bad1c4a3adfac99ca92b61f40ff5901fb052 Mon Sep 17 00:00:00 2001 From: atashakkor Date: Sat, 21 Dec 2024 15:40:30 +0330 Subject: [PATCH] Formatting. --- .../tests/test_postprocessor_voyageai-rerank.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/llama-index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/tests/test_postprocessor_voyageai-rerank.py b/llama-index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/tests/test_postprocessor_voyageai-rerank.py index 86fc3906b35c8..7cc4f0e9be3c3 100644 --- a/llama-index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/tests/test_postprocessor_voyageai-rerank.py +++ b/llama-index-integrations/postprocessor/llama-index-postprocessor-voyageai-rerank/tests/test_postprocessor_voyageai-rerank.py @@ -61,8 +61,11 @@ def test_rerank_construction_with_no_optional_kwargs(): assert reranker.top_n is None assert reranker.model == "rerank-2" + def test_rerank_construction_with_optional_kwargs(): - reranker = VoyageAIRerank(model="rerank-2", api_key="mock_api_key", top_n=10, truncation=False) + reranker = VoyageAIRerank( + model="rerank-2", api_key="mock_api_key", top_n=10, truncation=False + ) assert not reranker.truncation assert reranker.top_n == 10 - assert reranker.model == "rerank-2" \ No newline at end of file + assert reranker.model == "rerank-2"