pufanyi commited on
Commit
b62ae51
·
1 Parent(s): 19209dc

Refactor GET_DEFAULT_CHECKBOX function to remove "Model Name" column and handle exceptions in app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -128,9 +128,9 @@ def init_leaderboard(dataframes):
128
  if search_term:
129
  filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
130
  filtered_data.sort_values(by="Total", ascending=False, inplace=True)
131
- selected_columns.append("Model Name")
132
  selected_columns = [
133
- c for c in filtered_data.columns if c in selected_columns
134
  ]
135
  # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
136
  selected_data = filtered_data[selected_columns]
@@ -138,7 +138,7 @@ def init_leaderboard(dataframes):
138
  selected_data,
139
  type="pandas",
140
  # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
141
- )
142
 
143
  with gr.Row():
144
  refresh_button = gr.Button("Refresh")
@@ -151,9 +151,9 @@ def init_leaderboard(dataframes):
151
  concurrency_limit=20,
152
  )
153
 
154
- selected_subset.change(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=df)
155
- research_textbox.submit(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=df)
156
- selected_columns.change(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=df)
157
 
158
 
159
  def init_detailed_results():
 
128
  if search_term:
129
  filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
130
  filtered_data.sort_values(by="Total", ascending=False, inplace=True)
131
+ # selected_columns.append("Model Name")
132
  selected_columns = [
133
+ c for c in filtered_data.columns if c in selected_columns or c == "Model Name"
134
  ]
135
  # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
136
  selected_data = filtered_data[selected_columns]
 
138
  selected_data,
139
  type="pandas",
140
  # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
141
+ ), research_textbox, selected_columns
142
 
143
  with gr.Row():
144
  refresh_button = gr.Button("Refresh")
 
151
  concurrency_limit=20,
152
  )
153
 
154
+ selected_subset.change(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=[df, research_textbox, selected_columns])
155
+ research_textbox.submit(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=[df, research_textbox, selected_columns])
156
+ selected_columns.change(update_data, inputs=[selected_subset, research_textbox, selected_columns], outputs=[df, research_textbox, selected_columns])
157
 
158
 
159
  def init_detailed_results():