From 18849126747e02a530801d48686afd90b1bca2fc Mon Sep 17 00:00:00 2001 From: Hubert <42952108+yingfhu@users.noreply.github.com> Date: Fri, 24 Nov 2023 13:43:26 +0800 Subject: [PATCH] [Bug] fix icl eval with nested list (#632) --- opencompass/tasks/openicl_eval.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) 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),