djstrong commited on
Commit
b91291d
1 Parent(s): c1e0c7e
Files changed (2) hide show
  1. app.py +3 -3
  2. src/leaderboard/read_evals.py +2 -2
app.py CHANGED
@@ -79,8 +79,8 @@ def style_df(df: pd.DataFrame) -> Styler:
79
  leaderboard_df_styled = df.style.background_gradient(cmap="RdYlGn")
80
 
81
  inverted_colors_columns = ['rank']
82
- if "poleval2018_task3_test_10k" in df.columns:
83
- inverted_colors_columns.append("poleval2018_task3_test_10k")
84
  if '#Params (B)' in df.columns:
85
  inverted_colors_columns.append("#Params (B)")
86
 
@@ -88,7 +88,7 @@ def style_df(df: pd.DataFrame) -> Styler:
88
  rounding = {'#Params (B)': "{:.1f}", 'rank': "{:.0f}"}
89
  for task in Tasks:
90
  rounding[task.value.col_name] = "{:.2f}"
91
- for column_name in ["Average ⬆️", "Avg g", "Avg mc", "Average old", "Avg RAG"]:
92
  rounding[column_name] = "{:.2f}"
93
  leaderboard_df_styled = leaderboard_df_styled.format(rounding)
94
  return leaderboard_df_styled
 
79
  leaderboard_df_styled = df.style.background_gradient(cmap="RdYlGn")
80
 
81
  inverted_colors_columns = ['rank']
82
+ # if "poleval2018_task3_test_10k" in df.columns:
83
+ # inverted_colors_columns.append("poleval2018_task3_test_10k")
84
  if '#Params (B)' in df.columns:
85
  inverted_colors_columns.append("#Params (B)")
86
 
 
88
  rounding = {'#Params (B)': "{:.1f}", 'rank': "{:.0f}"}
89
  for task in Tasks:
90
  rounding[task.value.col_name] = "{:.2f}"
91
+ for column_name in ["Average ⬆️"]:
92
  rounding[column_name] = "{:.2f}"
93
  leaderboard_df_styled = leaderboard_df_styled.format(rounding)
94
  return leaderboard_df_styled
src/leaderboard/read_evals.py CHANGED
@@ -387,8 +387,8 @@ def get_raw_eval_results(results_path: str, requests_path: str, metadata) -> lis
387
  model_result_filepaths = []
388
 
389
  for root, _, files in os.walk(results_path):
390
- print(root)
391
- # if '_polish_pes_' not in root: continue
392
  # We should only have json files in model results
393
  if len(files) == 0 or any([not f.endswith(".json") for f in files]):
394
  continue
 
387
  model_result_filepaths = []
388
 
389
  for root, _, files in os.walk(results_path):
390
+ if '_polish_pes_' not in root: continue
391
+
392
  # We should only have json files in model results
393
  if len(files) == 0 or any([not f.endswith(".json") for f in files]):
394
  continue