From b8bf16e81c2cbeb67ba9b1945a75b0c2fad7111f Mon Sep 17 00:00:00 2001 From: Leymore Date: Tue, 5 Sep 2023 10:37:42 +0800 Subject: [PATCH] [Fix] zero retriever add default value (#361) --- opencompass/openicl/icl_retriever/icl_zero_retriever.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opencompass/openicl/icl_retriever/icl_zero_retriever.py b/opencompass/openicl/icl_retriever/icl_zero_retriever.py index c1d889c6..c63d6cf5 100644 --- a/opencompass/openicl/icl_retriever/icl_zero_retriever.py +++ b/opencompass/openicl/icl_retriever/icl_zero_retriever.py @@ -22,7 +22,7 @@ class ZeroRetriever(BaseRetriever): def __init__(self, dataset, ice_eos_token: Optional[str] = '') -> None: super().__init__(dataset, '', ice_eos_token, 0) - def retrieve(self, id_list: List[int]) -> List[List]: + def retrieve(self, id_list: List[int]=[]) -> List[List]: if id_list: get_logger().warning('id_list is not empty, but will be ignored.') rtr_idx_list = [[] for _ in range(len(self.test_ds))]