jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Fix] zero retriever add default value (#361)

Browse Source
This commit is contained in:
Leymore 2023-09-05 10:37:42 +08:00 committed by GitHub
parent 764c2f799a
commit b8bf16e81c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/openicl/icl_retriever/icl_zero_retriever.py
View File

@ -22,7 +22,7 @@ class ZeroRetriever(BaseRetriever):
def __init__(self, dataset, ice_eos_token: Optional[str] = '') -> None: def __init__(self, dataset, ice_eos_token: Optional[str] = '') -> None:
super().__init__(dataset, '', ice_eos_token, 0) 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: if id_list:
get_logger().warning('id_list is not empty, but will be ignored.') get_logger().warning('id_list is not empty, but will be ignored.')
rtr_idx_list = [[] for _ in range(len(self.test_ds))] rtr_idx_list = [[] for _ in range(len(self.test_ds))]