From ff7927ea48f1fbeea2ebf145dbcafed4588a6e38 Mon Sep 17 00:00:00 2001
From: Mor-Li <2568818204@qq.com>
Date: Sat, 26 Apr 2025 12:07:57 +0800
Subject: [PATCH] fix needlebench pic format

---
 opencompass/summarizers/needlebench.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/opencompass/summarizers/needlebench.py b/opencompass/summarizers/needlebench.py
index 0b3283b8..f2c744d0 100644
--- a/opencompass/summarizers/needlebench.py
+++ b/opencompass/summarizers/needlebench.py
@@ -370,7 +370,7 @@ def visualize(df_raw, save_path: str,model_name: str ,dataset_type:str):
 
         new_save_path = os.path.join(directory_path, new_filename)
 
-        plt.savefig(new_save_path, format='pdf', bbox_inches='tight', pad_inches=0)
+        plt.savefig(new_save_path, format='png', bbox_inches='tight', pad_inches=0)
         print(f'Saved: {new_save_path}')
 
         plt.close()