[Bug] fix icl eval with nested list (#632)

open-compass · Nov 24, 2023 · 1884912 · 1884912
1 parent 81b67e8
commit 1884912
Showing 1 changed file with 5 additions and 7 deletions.
diff --git 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),