From 99a2f108fbf3c266c4dfc2c4352968900751bbcf Mon Sep 17 00:00:00 2001 From: Thang Nguyen Date: Mon, 6 May 2024 10:24:38 +0700 Subject: [PATCH] add missing kawrgs when invoking retrieveRerankEnsemble --- dsp/primitives/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsp/primitives/search.py b/dsp/primitives/search.py index 0202098f0d..bfabed4955 100644 --- a/dsp/primitives/search.py +++ b/dsp/primitives/search.py @@ -49,7 +49,7 @@ def retrieveEnsemble(queries: list[str], k: int, by_prob: bool = True,**kwargs) if not dsp.settings.rm: raise AssertionError("No RM is loaded.") if dsp.settings.reranker: - return retrieveRerankEnsemble(queries, k) + return retrieveRerankEnsemble(queries, k, **kwargs) queries = [q for q in queries if q]