CMB-Clin: KeyError: 'solution'

[XiudingCai]

dict_keys(['id', 'title', 'description', 'QA_pairs'])
dict_keys(['id', 'title', 'description', 'QA_pairs'])
dict_keys(['question', 'answer'])
0%| | 0/5 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/abc/CMB/./src/generate_answers.py", line 210, in
main(args)
File "/home/abc/CMB/./src/generate_answers.py", line 63, in main
response, data = runner.generate_batch(batch)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/abc/miniconda3/envs/ane/lib/python3.12/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/home/abc/CMB/src/../workers/base.py", line 184, in generate_batch
batch, lines = self.prompt_wrapper.wrap_conv(batch)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/abc/CMB/src/../workers/base.py", line 317, in wrap_conv
collated, partial_qa = self.conv_collater(line)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/abc/CMB/src/../workers/deepseek_7b.py", line 73, in collate_conv
line += '答：{}\n'.format(conv['solution'])
~~~~^^^^^^^^^^^^
KeyError: 'solution'