rename column names

app.py

@@ -42,8 +42,7 @@ repo = snapshot_download(
 
 
 def avg_over_rewardbench_v2(dataframe_core):
-    # TODO: change this when I run on new data with different names
-    domain_cols = ["factuality", "precise if", "math", "safety", "chat", "ties"]
+    domain_cols = ["Factuality", "Precise IF", "Math", "Safety", "Focus", "Ties"]
     domain_weights = [1, 1, 1, 1, 1, 1]
     new_df = dataframe_core.copy()
 
@@ -59,18 +58,6 @@ def avg_over_rewardbench_v2(dataframe_core):
     # Rearrange columns for consistent output
     keep_columns = ["model", "model_type", "average"] + domain_cols
     new_df = new_df[keep_columns]
-
-    # TODO: update domain_cols and comment this out if final dataset version changes names
-    new_df = new_df.rename(
-        columns={
-            "factuality": "Factuality",
-            "precise if": "Precise IF",
-            "math": "Math",
-            "safety": "Safety",
-            "chat": "Focus",
-            "ties": "Ties",
-        }
-    )
     return new_df