mode

app.py

@@ -13,6 +13,7 @@ import plotly.graph_objects as go
 from apscheduler.schedulers.background import BackgroundScheduler
 import numpy as np
 from gradio.themes.utils import fonts, colors
+from dataclasses import fields, ColumnInfo
 
 from src.about import (
     CITATION_BUTTON_LABEL,
@@ -282,9 +283,13 @@ def init_leaderboard(dataframe, visible_columns=None):
 
     styler = display_df.style.set_properties(**{'text-align': 'right'})
 
+    # Get display names for headers
+    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]
+
     return gr.Dataframe(
         value=styler,
-        headers=visible_columns,
+        headers=display_headers,
         datatype=datatypes,  # Now using the correct format
         interactive=False,
         wrap=True,

src/display/utils.py

@@ -104,7 +104,7 @@ class GuardBenchColumn:
     ))
     model_type: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="model_type",
-        display_name="Type",
+        display_name="Access_Type",
         displayed_by_default=True
     ))
     submission_date: ColumnInfo = field(default_factory=lambda: ColumnInfo(
@@ -119,7 +119,7 @@ class GuardBenchColumn:
     ))
     guard_model_type: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="guard_model_type",
-        display_name="Guard Model Type",
+        display_name="Type",
         displayed_by_default=True
     ))
     base_model: ColumnInfo = field(default_factory=lambda: ColumnInfo(
@@ -296,7 +296,7 @@ class GuardBenchColumn:
     ))
     integral_score: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="integral_score",
-        display_name="Integral Score",
+        display_name="Integral_Score",
         type="number",
         displayed_by_default=True
     ))
@@ -304,13 +304,13 @@ class GuardBenchColumn:
     # Calculated overall metrics (renamed)
     macro_accuracy: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="macro_accuracy",
-        display_name="Macro Accuracy",
+        display_name="Macro_Accuracy",
         type="number",
         displayed_by_default=True
     ))
     macro_recall: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="macro_recall",
-        display_name="Macro Recall",
+        display_name="Macro_Recall",
         type="number",
         displayed_by_default=True
     ))
@@ -324,19 +324,19 @@ class GuardBenchColumn:
     # NEW Summary Metrics
     micro_avg_error_ratio: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="micro_avg_error_ratio",
-        display_name="Micro Error %",
+        display_name="Micro_Error",
         type="number",
         displayed_by_default=True
     ))
     micro_avg_runtime_ms: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="micro_avg_runtime_ms",
-        display_name="Micro Avg Time (ms)",
+        display_name="Micro_Avg_time_ms",
         type="number",
         displayed_by_default=True
     ))
     total_evals_count: ColumnInfo = field(default_factory=lambda: ColumnInfo(
         name="total_evals_count",
-        display_name="Total Evals Count",
+        display_name="Total_Count",
         type="number",
         displayed_by_default=True
     ))