Clean up

app.py

@@ -258,14 +258,13 @@ def toggle_all_categories(action: str) -> list[gr.CheckboxGroup]:
 TASK_AVG_NAME_MAP = {
     c.name: c.task_type.name for c in fields(AutoEvalColumn) if c.average and c.task_type != TaskType.AVG
 }
+AVG_COLUMNS = ["AVG"] + list(TASK_AVG_NAME_MAP.keys())
 
 
 def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
     df = ORIGINAL_DF[ORIGINAL_DF[AutoEvalColumn.row_id.name].isin(df_filtered[AutoEvalColumn.row_id.name])]
     df = df[df["#Params (B)"] > 0]
-    AVG_COLUMNS = ["AVG"] + list(TASK_AVG_NAME_MAP.keys())
     df = df[["model_name_for_query", "#Params (B)", "Few-shot"] + AVG_COLUMNS]
-    df[AVG_COLUMNS] = df[AVG_COLUMNS].astype(float)
     df = df.rename(columns={"model_name_for_query": "Model", "Few-shot": "n-shot"})
     df["model_name_without_org_name"] = df["Model"].str.split("/").str[-1] + " (" + df["n-shot"].astype(str) + "-shot)"
     df = pd.melt(
@@ -316,8 +315,7 @@ def plot_average_scores(df_filtered: pd.DataFrame) -> go.Figure:
     df = df[["model_name_for_query", "Few-shot"] + list(TASK_AVG_NAME_MAP.keys())]
     df = df.rename(columns={"model_name_for_query": "Model", "Few-shot": "n-shot"})
     df = df.rename(columns=TASK_AVG_NAME_MAP)
-    df["n-shot"] = df["n-shot"].astype(int)
-    df = df.set_index(["Model", "n-shot"]).astype(float)
+    df = df.set_index(["Model", "n-shot"])
 
     fig = go.Figure()
     for i, ((name, n_shot), row) in enumerate(df.iterrows()):