From a08931f21446a0dbf5c17024f6d1a564f3eed0b0 Mon Sep 17 00:00:00 2001 From: mqy004 <134621827+mqy004@users.noreply.github.com> Date: Fri, 26 Jul 2024 11:46:02 +0800 Subject: [PATCH] [Fix] origin_prompt should be None in llm-compression task (#1225) Co-authored-by: Qinyang Mou --- opencompass/tasks/openicl_eval.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py index 1fd9e7ed..1f9aff72 100644 --- a/opencompass/tasks/openicl_eval.py +++ b/opencompass/tasks/openicl_eval.py @@ -215,7 +215,12 @@ class OpenICLEvalTask(BaseTask): if self.output_column else None) preds['test_set'] = test_set if 'origin_prompt' not in preds: - preds['origin_prompt'] = [None for _ in range(len(pred_strs))] + try: + preds['origin_prompt'] = [ + None for _ in range(len(pred_strs)) + ] + except TypeError: + preds['origin_prompt'] = None preds = { k: preds[k] for k in signature(icl_evaluator.score).parameters