Skip to content

Commit 871c3fc

Browse files
author
Zhe Yu
committed
some cleanup
1 parent ed4b3bd commit 871c3fc

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

tests/subcommands/query/test_reranker.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -113,10 +113,7 @@ def test_naive_reranker_handles_none_path(config, query_result):
113113

114114

115115
@patch("sentence_transformers.CrossEncoder")
116-
def test_cross_encoder_reranker_initialization(
117-
mock_cross_encoder: MagicMock, config, query_chunks
118-
):
119-
"""Test initialization of CrossEncoderReranker"""
116+
def test_cross_encoder_reranker_initialization(mock_cross_encoder: MagicMock, config):
120117
reranker = CrossEncoderReranker(config)
121118

122119
# Verify constructor was called with correct parameters
@@ -128,8 +125,6 @@ def test_cross_encoder_reranker_initialization(
128125
def test_cross_encoder_reranker_rerank(
129126
mock_cross_encoder, config, query_result, query_chunks
130127
):
131-
"""Test reranking with CrossEncoderReranker"""
132-
# Setup mock model
133128
mock_model = MagicMock()
134129
mock_cross_encoder.return_value = mock_model
135130

0 commit comments

Comments
 (0)