Skip to content

Commit cf49f8b

Browse files
author
Zhe Yu
committed
fix broken tests (for github actions)
1 parent b6e9aad commit cf49f8b

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

tests/subcommands/query/test_reranker.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -262,13 +262,11 @@ def test_get_reranker():
262262
), "configs.reranker_params should fallback to default params."
263263

264264

265-
def test_supported_rerankers_initialization():
265+
def test_supported_rerankers_initialization(config, naive_reranker_conf):
266266
"""Test that __supported_rerankers contains the expected default rerankers"""
267267

268-
assert isinstance(
269-
get_reranker(Config(reranker="CrossEncoderReranker")), CrossEncoderReranker
270-
)
271-
assert isinstance(get_reranker(Config(reranker="NaiveReranker")), NaiveReranker)
268+
assert isinstance(get_reranker(config), CrossEncoderReranker)
269+
assert isinstance(get_reranker(naive_reranker_conf), NaiveReranker)
272270
assert len(get_available_rerankers()) == 2
273271

274272

0 commit comments

Comments
 (0)