mode

app.py

@@ -290,20 +290,18 @@ def init_leaderboard(dataframe, visible_columns=None):
             # Format floats to exactly 3 decimal places, preserving trailing zeros
             display_df[col] = display_df[col].apply(lambda x: f"{x:.3f}" if pd.notna(x) else None)
 
-    styler = display_df.style.set_properties(**{'text-align': 'right'})
-
-    # Get display names for headers
+    # Get display names and create mapping
     column_info_map = {f.name: getattr(GUARDBENCH_COLUMN, f.name) for f in fields(GUARDBENCH_COLUMN)}
-    display_headers = [column_info_map.get(col, ColumnInfo(col, col)).display_name for col in visible_columns]
-    print("--------------------------------")
-    print("display_headers", display_headers)
-    print("visible_columns", visible_columns)
-    print("column_info_map", column_info_map)
-    print("--------------------------------")
+    column_mapping = {col: column_info_map.get(col, ColumnInfo(col, col)).display_name for col in visible_columns}
+
+    # Rename columns in the DataFrame
+    display_df.rename(columns=column_mapping, inplace=True)
+
+    # Apply styling - note: styling might need adjustment if it relies on column names
+    styler = display_df.style.set_properties(**{'text-align': 'right'})
 
     return gr.Dataframe(
-        value=styler,
-        headers=display_headers,
+        value=styler, # Pass the DataFrame with renamed columns
         datatype=datatypes,  # Now using the correct format
         interactive=False,
         wrap=True,