pufanyi commited on
Commit
de5feeb
·
1 Parent(s): d3ac26d

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 +5 -1
app.py CHANGED
@@ -129,7 +129,11 @@ def init_leaderboard(dataframes):
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 = list(selected_columns)
 
 
 
 
133
  selected_columns = [
134
  c for c in filtered_data.columns if c in selected_columns or c == "Model Name"
135
  ]
 
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
+ try:
133
+ selected_columns = selected_columns.choices
134
+ except Exception as e:
135
+ print(e)
136
+ selected_columns = selected_columns
137
  selected_columns = [
138
  c for c in filtered_data.columns if c in selected_columns or c == "Model Name"
139
  ]