Refactor GET_DEFAULT_CHECKBOX function to accept a subset parameter

app.py

@@ -78,8 +78,8 @@ def GET_DEFAULT_TEXTBOX():
     return gr.Textbox("", placeholder="🔍 Search Models... [press enter]", label="Filter Models by Name")
 
 
-def GET_DEFAULT_CHECKBOX():
-    choices = list(LEADERBOARD_DF.columns)
+def GET_DEFAULT_CHECKBOX(subset):
+    choices = list(LEADERBOARD_DF[subset].columns)
     choices.remove("Model Name")
     # print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
     return gr.CheckboxGroup(
@@ -95,7 +95,7 @@ def init_leaderboard(dataframes):
     with gr.Row():
         selected_subset = gr.Dropdown(choices=subsets, label="Select Dataset Subset", value=NEWEST_VERSION)
         research_textbox = GET_DEFAULT_TEXTBOX()
-        selected_columns = GET_DEFAULT_CHECKBOX()
+        selected_columns = GET_DEFAULT_CHECKBOX(NEWEST_VERSION)
 
     data = dataframes[NEWEST_VERSION]
 
@@ -107,10 +107,10 @@ def init_leaderboard(dataframes):
         global LEADERBOARD_DF
         LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO)
         # default_columns = [c.name for c in fields(AutoEvalColumn) if c.displayed_by_default]
-        default_columns = list(LEADERBOARD_DF.columns)
+        default_columns = list(LEADERBOARD_DF[subset].columns)
         default_columns.remove("Model Name")
 
-        return update_data(subset, None, default_columns), GET_DEFAULT_TEXTBOX(), GET_DEFAULT_CHECKBOX()
+        return update_data(subset, None, default_columns), GET_DEFAULT_TEXTBOX(), GET_DEFAULT_CHECKBOX(subset)
 
     def update_data(subset, search_term, selected_columns):
         print("Subset:", subset)