update populate

src/populate.py

@@ -6,7 +6,7 @@ import json
 import random
 
 from src.display.utils import COLUMNS, EVAL_COLS, Tasks
-from src.envs import EVAL_RESULTS_PATH, FIXED_QUESTIONS_FILE  # Define FIXED_QUESTIONS_FILE in envs.py
+from src.envs import EVAL_RESULTS_PATH, FIXED_QUESTIONS_FILE  # Ensure FIXED_QUESTIONS_FILE is defined in envs.py
 
 def get_leaderboard_df(eval_results_path, eval_requests_path, cols, benchmark_cols):
     # Initialize an empty DataFrame
@@ -14,12 +14,16 @@ def get_leaderboard_df(eval_results_path, eval_requests_path, cols, benchmark_co
 
     # Load evaluation results from JSON files
     if os.path.exists(eval_results_path):
-        result_files = [os.path.join(eval_results_path, f) for f in os.listdir(eval_results_path) if f.endswith('.json')]
+        result_files = [
+            os.path.join(eval_results_path, f)
+            for f in os.listdir(eval_results_path)
+            if f.endswith('.json')
+        ]
         data_list = []
         for file in result_files:
             with open(file, 'r') as f:
                 data = json.load(f)
-                # Flatten the JSON structure if needed
+                # Flatten the JSON structure
                 flattened_data = {}
                 flattened_data.update(data.get('config', {}))
                 flattened_data.update(data.get('results', {}))
@@ -27,6 +31,10 @@ def get_leaderboard_df(eval_results_path, eval_requests_path, cols, benchmark_co
         if data_list:
             df = pd.DataFrame(data_list)
 
+    # Rename 'model_name' to 'model' if 'model' is missing
+    if 'model' not in df.columns and 'model_name' in df.columns:
+        df.rename(columns={'model_name': 'model'}, inplace=True)
+
     # Ensure DataFrame has all columns
     for col in cols:
         if col not in df.columns:
@@ -46,7 +54,11 @@ def get_evaluation_queue_df(eval_requests_path, eval_cols):
 
     # Load evaluation requests from JSON files
     if os.path.exists(eval_requests_path):
-        request_files = [os.path.join(eval_requests_path, f) for f in os.listdir(eval_requests_path) if f.endswith('.json')]
+        request_files = [
+            os.path.join(eval_requests_path, f)
+            for f in os.listdir(eval_requests_path)
+            if f.endswith('.json')
+        ]
         data_list = []
         for file in request_files:
             with open(file, 'r') as f: