Skip to content

GitLab

Unverified Commit 880b34e7 authored by Leymore's avatar Leymore Committed by GitHub
Browse files

[Fix] Quick lint fix (#362) 

* add default value

* lint fix

* use None
parent 5d75c1bb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
opencompass/openicl/icl_retriever/icl_zero_retriever.py
View file @ 880b34e7
...@@ -22,8 +22,8 @@ class ZeroRetriever(BaseRetriever): ...@@ -22,8 +22,8 @@ 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] = None) -> List[List]:
if id_list: if id_list is not None:
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))]
return rtr_idx_list 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