Repository: open-compass/opencompass
Issue: [Bug] CMB数据集报错
Root cause: Validation Gap

### Prerequisite

- [X] I have searched [Issues](https://github.com/open-compass/opencompass/issues/) and [Discussions](https://github.com/open-compass/opencompass/discussions) but cannot get the expected help.
- [X] The bug has not been fixed in the [latest version](https://github.com/open-compass/opencompass).

### Type

I'm evaluating with the officially supported tasks/models/datasets.

### Environment

{'CUDA available': True,
 'CUDA_HOME': '/usr/local/cuda-12.2',
 'GCC': 'gcc (Ubuntu 12.3.0-1ubuntu1~22.04) 12.3.0',
 'GPU 0,1': 'NVIDIA GeForce RTX 4090',
 'MMEngine': '0.10.4',
 'MUSA available': False,
 'NVCC': 'Cuda compilation tools, release 12.2, V12.2.140',
 'OpenCV': '4.9.0',
 'PyTorch': '2.2.2',
 'PyTorch compiling details': 'PyTorch built with:\n'
                              '  - GCC 9.3\n'
                              '  - C++ Version: 201703\n'
                              '  - Intel(R) oneAPI Math Kernel Library Version '
                              '2023.1-Product Build 20230303 for Intel(R) 64 '
                              'architecture applications\n'
                              '  - Intel(R) MKL-DNN v3.3.2 (Git Hash '
                              '2dc95a2ad0841e29db8b22fbccaf3e5da7992b01)\n'
                              '  - OpenMP 201511 (a.k.a. OpenMP 4.5)\n'
                              '  - LAPACK is enabled (usually provided by '
                              'MKL)\n'
                              '  - NNPACK is enabled\n'
                              '  - CPU capability usage: AVX2\n'
                              '  - CUDA Runtime 12.1\n'
                              '  - NVCC architecture flags: '
                              '-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_61,code=sm_61;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_80,code=sm_80;-gencode;arch=compute_86,code=sm_86;-gencode;arch=compute_90,code=sm_90\n'
                              '  - CuDNN 8.9.2\n'
                              '  - Magma 2.6.1\n'
                              '  - Build settings: BLAS_INFO=mkl, '
                              'BUILD_TYPE=Release, CUDA_VERSION=12.1, '
                              'CUDNN_VERSION=8.9.2, '
                              'CXX_COMPILER=/opt/rh/devtoolset-9/root/usr/bin/c++, '
                              'CXX_FLAGS= -D_GLIBCXX_USE_CXX11_ABI=0 '
                              '-fabi-version=11 -fvisibility-inlines-hidden '
                              '-DUSE_PTHREADPOOL -DNDEBUG -DUSE_KINETO '
                              '-DLIBKINETO_NOROCTRACER -DUSE_FBGEMM '
                              '-DUSE_QNNPACK -DUSE_PYTORCH_QNNPACK '
                              '-DUSE_XNNPACK -DSYMBOLICATE_MOBILE_DEBUG_HANDLE '
                              '-O2 -fPIC -Wall -Wextra -Werror=return-type '
                              '-Werror=non-virtual-dtor -Werror=bool-operation '
                              '-Wnarrowing -Wno-missing-field-initializers '
                              '-Wno-type-limits -Wno-array-bounds '
                              '-Wno-unknown-pragmas -Wno-unused-parameter '
                              '-Wno-unused-function -Wno-unused-result '
                              '-Wno-strict-overflow -Wno-strict-aliasing '
                              '-Wno-stringop-overflow -Wsuggest-override '
                              '-Wno-psabi -Wno-error=pedantic '
                              '-Wno-error=old-style-cast -Wno-missing-braces '
                              '-fdiagnostics-color=always -faligned-new '
                              '-Wno-unused-but-set-variable '
                              '-Wno-maybe-uninitialized -fno-math-errno '
                              '-fno-trapping-math -Werror=format '
                              '-Wno-stringop-overflow, LAPACK_INFO=mkl, '
                              'PERF_WITH_AVX=1, PERF_WITH_AVX2=1, '
                              'PERF_WITH_AVX512=1, TORCH_VERSION=2.2.2, '
                              'USE_CUDA=ON, USE_CUDNN=ON, USE_EXCEPTION_PTR=1, '
                              'USE_GFLAGS=OFF, USE_GLOG=OFF, USE_MKL=ON, '
                              'USE_MKLDNN=ON, USE_MPI=OFF, USE_NCCL=ON, '
                              'USE_NNPACK=ON, USE_OPENMP=ON, USE_ROCM=OFF, '
                              'USE_ROCM_KERNEL_ASSERT=OFF, \n',
 'Python': '3.10.14 (main, Mar 21 2024, 16:24:04) [GCC 11.2.0]',
 'TorchVision': '0.17.2',
 'numpy_random_seed': 2147483648,
 'opencompass': '0.2.4+8fe7b27',
 'sys.platform': 'linux'}


### Reproduces the problem - code/configuration sample

  python run.py --datasets cmb_gen --hf-path /opt/models/Qwen1.5-0.5B-Chat/ --tokenizer-path /opt/models/Qwen1.5-0.5B-Chat/ --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True --model-kwargs trust_remote_code=True device_map='auto' --max-seq-len 1024 --max-out-len 16 --batch-size 2 --num-gpus 1 --debug

### Reproduces the problem - command or script

  python run.py --datasets cmb_gen --hf-path /opt/models/Qwen1.5-0.5B-Chat/ --tokenizer-path /opt/models/Qwen1.5-0.5B-Chat/ --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True --model-kwargs trust_remote_code=True device_map='auto' --max-seq-len 1024 --max-out-len 16 --batch-size 2 --num-gpus 1 --debug

### Reproduces the problem - error message

04/23 17:05:13 - OpenCompass - INFO - Loading cmb_gen: configs/datasets/cmb/cmb_gen.py
04/23 17:05:13 - OpenCompass - INFO - Loading example: configs/summarizers/example.py
04/23 17:05:13 - OpenCompass - WARNING - SlurmRunner is not used, so the partition argument is ignored.
04/23 17:05:13 - OpenCompass - DEBUG - Modules of opencompass's partitioner registry have been automatically imported from opencompass.partitioners
04/23 17:05:13 - OpenCompass - DEBUG - Get class `SizePartitioner` from "partitioner" registry in "opencompass"
04/23 17:05:13 - OpenCompass - DEBUG - An `SizePartitioner` instance is built from registry, and its implementation can be found in opencompass.partitioners.size
04/23 17:05:13 - OpenCompass - DEBUG - Key eval.runner.task.judge_cfg not found in config, ignored.
04/23 17:05:13 - OpenCompass - DEBUG - Key eval.runner.task.dump_details not found in config, ignored.
04/23 17:05:13 - OpenCompass - DEBUG - Key eval.given_pred not found in config, ignored.
04/23 17:05:13 - OpenCompass - DEBUG - Additional config: {}
04/23 17:05:13 - OpenCompass - INFO - Partitioned into 7 tasks.
04/23 17:05:13 - OpenCompass - DEBUG - Task 0: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_0]
04/23 17:05:13 - OpenCompass - DEBUG - Task 1: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_1]
04/23 17:05:13 - OpenCompass - DEBUG - Task 2: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_2]
04/23 17:05:13 - OpenCompass - DEBUG - Task 3: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_3]
04/23 17:05:13 - OpenCompass - DEBUG - Task 4: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_4]
04/23 17:05:13 - OpenCompass - DEBUG - Task 5: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_5]
04/23 17:05:13 - OpenCompass - DEBUG - Task 6: [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb]
04/23 17:05:13 - OpenCompass - DEBUG - Modules of opencompass's runner registry have been automatically imported from opencompass.runners
04/23 17:05:13 - OpenCompass - DEBUG - Get class `LocalRunner` from "runner" registry in "opencompass"
04/23 17:05:13 - OpenCompass - DEBUG - An `LocalRunner` instance is built from registry, and its implementation can be found in opencompass.runners.local
04/23 17:05:13 - OpenCompass - DEBUG - Modules of opencompass's task registry have been automatically imported from opencompass.tasks
04/23 17:05:13 - OpenCompass - DEBUG - Get class `OpenICLInferTask` from "task" registry in "opencompass"
04/23 17:05:13 - OpenCompass - DEBUG - An `OpenICLInferTask` instance is built from registry, and its implementation can be found in opencompass.tasks.openicl_infer
04/23 17:05:14 - OpenCompass - WARNING - Only use 1 GPUs for total 2 available GPUs in debug mode.
Error: mkl-service + Intel(R) MKL: MKL_THREADING_LAYER=INTEL is incompatible with libgomp.so.1 library.
	Try to import numpy first or set the threading layer accordingly. Set MKL_SERVICE_FORCE_INTEL to force it.
04/23 17:05:15 - OpenCompass - INFO - Task [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_0]
Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
04/23 17:05:17 - OpenCompass - INFO - Start inferencing [opencompass.models.huggingface.HuggingFace_models_Qwen1.5-0.5B-Chat/cmb_test_0]
Traceback (most recent call last):
  File "/root/opencompass/opencompass/tasks/openicl_infer.py", line 160, in <module>
    inferencer.run()
  File "/root/opencompass/opencompass/tasks/openicl_infer.py", line 88, in run
    self._inference()
  File "/root/opencompass/opencompass/tasks/openicl_infer.py", line 133, in _inference
    inferencer.inference(retriever,
  File "/root/opencompass/opencompass/openicl/icl_inferencer/icl_gen_inferencer.py", line 108, in inference
    gold_ans = ds_reader.dataset['test'][ds_reader.output_column]
  File "/root/anaconda3/envs/opencompass/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2861, in __getitem__
    return self._getitem(key)
  File "/root/anaconda3/envs/opencompass/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2845, in _getitem
    pa_subtable = query_table(self._data, key, indices=self._indices)
  File "/root/anaconda3/envs/opencompass/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 584, in query_table
    _check_valid_column_key(key, table.column_names)
  File "/root/anaconda3/envs/opencompass/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 521, in _check_valid_column_key
    raise KeyError(f"Column {key} not in the dataset. Current columns in the dataset: {columns}")
KeyError: "Column answer not in the dataset. Current columns in the dataset: ['id', 'exam_type', 'exam_class', 'exam_subject', 'question', 'question_type', 'option', 'option_str']"


### Other information

_No response_

hi I have fixed this problem in here https://github.com/open-compass/opencompass/pull/1106