Spaces:
AIR-Bench
/
Running on CPU Upgrade

nan commited on
Commit
7beeb69
1 Parent(s): fbcdfeb

chore: clean up

Browse files
Files changed (3) hide show
  1. app.py +4 -9
  2. src/display/gradio_listener.py +1 -1
  3. src/utils.py +4 -3
app.py CHANGED
@@ -11,11 +11,12 @@ from src.about import (
11
  from src.benchmarks import DOMAIN_COLS_QA, LANG_COLS_QA, DOMAIN_COLS_LONG_DOC, LANG_COLS_LONG_DOC, METRIC_LIST, \
12
  DEFAULT_METRIC
13
  from src.display.css_html_js import custom_css
14
- from src.display.utils import COL_NAME_IS_ANONYMOUS, COL_NAME_REVISION, COL_NAME_TIMESTAMP, COL_NAME_RERANKING_MODEL, COL_NAME_RANK, COL_NAME_AVG
15
  from src.envs import API, EVAL_RESULTS_PATH, REPO_ID, RESULTS_REPO, TOKEN
16
  from src.read_evals import get_raw_eval_results, get_leaderboard_df
17
- from src.utils import update_metric, upload_file, get_default_cols, submit_results
18
- from src.display.gradio_formatting import get_version_dropdown, get_search_bar, get_reranking_dropdown, get_noreranker_button, get_metric_dropdown, get_domain_dropdown, get_language_dropdown, get_anonymous_checkbox, get_revision_and_ts_checkbox, get_leaderboard_table, get_noreranking_dropdown
 
19
  from src.display.gradio_listener import set_listeners
20
 
21
  def restart_space():
@@ -83,12 +84,6 @@ def update_metric_long_doc(
83
 
84
  demo = gr.Blocks(css=custom_css)
85
 
86
-
87
- def reset_rank(df):
88
- df[COL_NAME_RANK] = df[COL_NAME_AVG].rank(ascending=False, method="min")
89
- return df
90
-
91
-
92
  with demo:
93
  gr.HTML(TITLE)
94
  gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
11
  from src.benchmarks import DOMAIN_COLS_QA, LANG_COLS_QA, DOMAIN_COLS_LONG_DOC, LANG_COLS_LONG_DOC, METRIC_LIST, \
12
  DEFAULT_METRIC
13
  from src.display.css_html_js import custom_css
14
+ from src.display.utils import COL_NAME_IS_ANONYMOUS, COL_NAME_REVISION, COL_NAME_TIMESTAMP, COL_NAME_RERANKING_MODEL
15
  from src.envs import API, EVAL_RESULTS_PATH, REPO_ID, RESULTS_REPO, TOKEN
16
  from src.read_evals import get_raw_eval_results, get_leaderboard_df
17
+ from src.utils import update_metric, upload_file, get_default_cols, submit_results, reset_rank
18
+ from src.display.gradio_formatting import get_version_dropdown, get_search_bar, get_reranking_dropdown, \
19
+ get_metric_dropdown, get_domain_dropdown, get_language_dropdown, get_anonymous_checkbox, get_revision_and_ts_checkbox, get_leaderboard_table, get_noreranking_dropdown
20
  from src.display.gradio_listener import set_listeners
21
 
22
  def restart_space():
 
84
 
85
  demo = gr.Blocks(css=custom_css)
86
 
 
 
 
 
 
 
87
  with demo:
88
  gr.HTML(TITLE)
89
  gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
src/display/gradio_listener.py CHANGED
@@ -1,4 +1,4 @@
1
- from src.utils import update_table, update_table_long_doc, clear_reranking_selections
2
 
3
 
4
  def set_listeners(
 
1
+ from src.utils import update_table, update_table_long_doc
2
 
3
 
4
  def set_listeners(
src/utils.py CHANGED
@@ -120,7 +120,7 @@ def select_columns(
120
  filtered_df[COL_NAME_AVG] = filtered_df[selected_cols].apply(calculate_mean, axis=1).round(decimals=2)
121
  filtered_df.sort_values(by=[COL_NAME_AVG], ascending=False, inplace=True)
122
  filtered_df.reset_index(inplace=True, drop=True)
123
- filtered_df[COL_NAME_RANK] = filtered_df[COL_NAME_AVG].rank(ascending=False, method="min")
124
 
125
  return filtered_df
126
 
@@ -312,5 +312,6 @@ def submit_results(
312
  )
313
 
314
 
315
- def clear_reranking_selections():
316
- return ["NoReranker",]
 
 
120
  filtered_df[COL_NAME_AVG] = filtered_df[selected_cols].apply(calculate_mean, axis=1).round(decimals=2)
121
  filtered_df.sort_values(by=[COL_NAME_AVG], ascending=False, inplace=True)
122
  filtered_df.reset_index(inplace=True, drop=True)
123
+ filtered_df = reset_rank(filtered_df)
124
 
125
  return filtered_df
126
 
 
312
  )
313
 
314
 
315
+ def reset_rank(df):
316
+ df[COL_NAME_RANK] = df[COL_NAME_AVG].rank(ascending=False, method="min")
317
+ return df