apsys commited on
Commit
61ccc5a
·
1 Parent(s): 67f06f8
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -290,7 +290,7 @@ def init_leaderboard(dataframe, visible_columns=None):
290
  # Format floats to exactly 3 decimal places, preserving trailing zeros
291
  display_df[col] = display_df[col].apply(lambda x: f"{x:.3f}" if pd.notna(x) else None)
292
 
293
- # Get display names and create mapping
294
  column_info_map = {f.name: getattr(GUARDBENCH_COLUMN, f.name) for f in fields(GUARDBENCH_COLUMN)}
295
  column_mapping = {col: column_info_map.get(col, ColumnInfo(col, col)).display_name for col in visible_columns}
296
 
@@ -301,8 +301,8 @@ def init_leaderboard(dataframe, visible_columns=None):
301
  styler = display_df.style.set_properties(**{'text-align': 'right'})
302
 
303
  return gr.Dataframe(
304
- value=styler, # Pass the DataFrame with renamed columns
305
- datatype=datatypes, # Now using the correct format
306
  interactive=False,
307
  wrap=True,
308
  elem_id="leaderboard-table",
@@ -501,7 +501,7 @@ def create_performance_plot(selected_models, category, metric="f1_binary", versi
501
  """
502
  Create a radar plot comparing model performance for selected models.
503
  """
504
- if category == "Overall Performance":
505
  df = get_leaderboard_df(version=version)
506
  else:
507
  df = get_category_leaderboard_df(category, version=version)
 
290
  # Format floats to exactly 3 decimal places, preserving trailing zeros
291
  display_df[col] = display_df[col].apply(lambda x: f"{x:.3f}" if pd.notna(x) else None)
292
 
293
+
294
  column_info_map = {f.name: getattr(GUARDBENCH_COLUMN, f.name) for f in fields(GUARDBENCH_COLUMN)}
295
  column_mapping = {col: column_info_map.get(col, ColumnInfo(col, col)).display_name for col in visible_columns}
296
 
 
301
  styler = display_df.style.set_properties(**{'text-align': 'right'})
302
 
303
  return gr.Dataframe(
304
+ value=styler,
305
+ datatype=datatypes,
306
  interactive=False,
307
  wrap=True,
308
  elem_id="leaderboard-table",
 
501
  """
502
  Create a radar plot comparing model performance for selected models.
503
  """
504
+ if category == "All Results":
505
  df = get_leaderboard_df(version=version)
506
  else:
507
  df = get_category_leaderboard_df(category, version=version)