Skip to content

GitLab

Unverified Commit 18849126 authored by Hubert's avatar Hubert Committed by GitHub
Browse files

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

parent 81b67e8d
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 7 deletions
opencompass/tasks/openicl_eval.py
View file @ 18849126
...@@ -135,13 +135,11 @@ class OpenICLEvalTask(BaseTask): ...@@ -135,13 +135,11 @@ class OpenICLEvalTask(BaseTask):
'The prediction for Self-Consistency' 'The prediction for Self-Consistency'
'must be list.') 'must be list.')
if pred_list_flag: if pred_list_flag:
for pred in pred_strs: pred_strs = [[
pred_strs.append([ self._extract_role_pred(_pred, role.get('begin', None),
self._extract_role_pred(_pred,
role.get('begin', None),
role.get('end', None)) role.get('end', None))
for _pred in pred for _pred in pred
]) ] for pred in pred_strs]
else: else:
pred_strs = [ pred_strs = [
self._extract_role_pred(pred, role.get('begin', None), self._extract_role_pred(pred, role.get('begin', None),
......
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