/home/wumao/opencompass-main/opencompass/summarizers/needlebench.py 的sizes是不是少了一个128k [Bug] [Feature]

[1518630367]

Describe the feature

导致报错：
Traceback (most recent call last):
File "/home/wumao/opencompass-main/run.py", line 4, in
main()
File "/home/wumao/opencompass-main/opencompass/cli/main.py", line 354, in main
summarizer.summarize(time_str=cfg_time_str)
File "/home/wumao/opencompass-main/opencompass/summarizers/needlebench.py", line 540, in summarize
save_results_to_plots(output_path)
File "/home/wumao/opencompass-main/opencompass/summarizers/needlebench.py", line 215, in save_results_to_plots
folder_path = os.path.join(plot_path, dataset_mapping_dict[dataset_abbr])
KeyError: '2needle_en_128k'

Will you implement it?

• I would like to implement this feature and create a PR!

[DseidLi]

Thank you for your careful attention to this matter! I have updated the code accordingly in #1143.