set precision

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
 from huggingface_hub import snapshot_download
+from pandas.io.formats.style import Styler
 
 from src.about import (
     CITATION_BUTTON_LABEL,
@@ -11,7 +12,7 @@ from src.about import (
     EVALUATION_QUEUE_TEXT,
     INTRODUCTION_TEXT,
     LLM_BENCHMARKS_TEXT,
-    TITLE,
+    TITLE, Tasks,
 )
 from src.display.css_html_js import custom_css
 from src.display.utils import (
@@ -66,6 +67,15 @@ leaderboard_df = original_df.copy()
     pending_eval_queue_df,
 ) = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
 
+def style_df(df: pd.DataFrame) -> Styler:
+    leaderboard_df_styled = df.style.background_gradient(cmap="viridis")
+    rounding = {'#Params (B)': "{:.1f}"}
+    for task in Tasks:
+        rounding[task.value.col_name] = "{:.2f}"
+    for column_name in ["Average ⬆️", "Avg g", "Avg mc"]:
+        rounding[column_name] = "{:.2f}"
+    leaderboard_df_styled = leaderboard_df_styled.format(rounding)
+    return leaderboard_df_styled
 
 # Searching and filtering
 def update_table(
@@ -87,10 +97,7 @@ def update_table(
 
     df = df.replace({'': np.nan})
 
-    leaderboard_df_styled = df.style.background_gradient(cmap="viridis")
-    # leaderboard_df_styled = leaderboard_df_styled.format("{:.2f}")
-    return leaderboard_df_styled
-
+    return style_df(df)
 
 def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False))]
@@ -226,8 +233,9 @@ with demo:
                     + shown_columns.value
                     + [AutoEvalColumn.dummy.name]
                 ]
-            leaderboard_df_styled = leaderboard_table_value.style.background_gradient(cmap="viridis")
-            # leaderboard_df_styled = leaderboard_df_styled.format("{:.2f}")
+            leaderboard_df_styled=style_df(leaderboard_table_value)
+
+
             leaderboard_df_styled.precision = 2
 
             leaderboard_table = gr.components.Dataframe(