diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py index f0e4b2d61..d39816f7b 100644 --- a/opencompass/tasks/openicl_eval.py +++ b/opencompass/tasks/openicl_eval.py @@ -135,13 +135,11 @@ def postprocess(sample): 'The prediction for Self-Consistency' 'must be list.') if pred_list_flag: - for pred in pred_strs: - pred_strs.append([ - self._extract_role_pred(_pred, - role.get('begin', None), - role.get('end', None)) - for _pred in pred - ]) + pred_strs = [[ + self._extract_role_pred(_pred, role.get('begin', None), + role.get('end', None)) + for _pred in pred + ] for pred in pred_strs] else: pred_strs = [ self._extract_role_pred(pred, role.get('begin', None),