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