Unverified Commit 8774465a authored by Leymore's avatar Leymore Committed by GitHub
Browse files

[Enhancement] ignore ZeroRetriever error when id_list provided (#340)

parent f2dd98ca
......@@ -4,6 +4,7 @@ from typing import List, Optional
from opencompass.openicl.icl_retriever import BaseRetriever
from opencompass.registry import ICL_RETRIEVERS
from opencompass.utils.logging import get_logger
@ICL_RETRIEVERS.register_module()
......@@ -21,6 +22,8 @@ class ZeroRetriever(BaseRetriever):
def __init__(self, dataset, ice_eos_token: Optional[str] = '') -> None:
super().__init__(dataset, '', ice_eos_token, 0)
def retrieve(self) -> 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))]
return rtr_idx_list
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment