python data2json.py --K 10 --temp_type simple --set test --dataset ml-1m 报错

[yearnforBCI]

(ReLLaVicuna) root@autodl-container-a8e51198ae-fbe0d52e:/autodl-tmp/ReLLa-main# python data2json.py --K 5 --temp_type simple --set train --dataset ml-1m
Namespace(K=5, temp_type='simple', dataset='ml-1m', set='train')
0%| | 0/776007 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/root/autodl-tmp/ReLLa-main/data2json.py", line 81, in
for msg, idx in zip(msg_iter, df.index):
File "/root/autodl-tmp/ReLLa-main/load_prompt_ml1m.py", line 131, in zero_shot_get_prompt
cur_temp = row_to_prompt(index, df, K, id_to_title, temp_type)
File "/root/autodl-tmp/ReLLa-main/load_prompt_ml1m.py", line 198, in row_to_prompt
assert key in row.keys(), "Key name error."
AssertionError: Key name error.
(ReLLaVicuna) root@autodl-container-a8e51198ae-fbe0d52e:/autodl-tmp/ReLLa-main#

你好，请问解决方法是什么