From 8774465a8ffd6ef8aaee75c1273ddbbd2eb1887e Mon Sep 17 00:00:00 2001 From: Leymore Date: Mon, 4 Sep 2023 11:12:16 +0800 Subject: [PATCH] [Enhancement] ignore ZeroRetriever error when id_list provided (#340) --- opencompass/openicl/icl_retriever/icl_zero_retriever.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/opencompass/openicl/icl_retriever/icl_zero_retriever.py b/opencompass/openicl/icl_retriever/icl_zero_retriever.py index 7ffb01c8..c1d889c6 100644 --- a/opencompass/openicl/icl_retriever/icl_zero_retriever.py +++ b/opencompass/openicl/icl_retriever/icl_zero_retriever.py @@ -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