LucasLima commited on
Commit
5efb8b6
·
verified ·
1 Parent(s): 9607afa

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +2 -11
src/populate.py CHANGED
@@ -55,19 +55,10 @@ def get_leaderboard_df(results_path: str = None, requests_path: str = None, cols
55
  if AutoEvalColumn.average.name in df.columns:
56
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
57
 
58
- # Selecionar e arredondar colunas
59
- # Usar as colunas definidas em AutoEvalColumn que existem no DataFrame
60
- defined_cols = [c.name for c in fields(AutoEvalColumn)]
61
- cols_to_keep = [col for col in defined_cols if col in df.columns]
62
-
63
- # Adicionar colunas que podem não estar em AutoEvalColumn mas estavam no DF original
64
- # (Ex: colunas de tasks individuais que foram usadas para cálculo mas não definidas explicitamente para exibição inicial)
65
- original_cols_needed_for_display = set(cols) if cols else set() # cols passados para a função
66
- final_cols = sorted(list(set(cols_to_keep) | (original_cols_needed_for_display & set(df.columns))))
67
-
68
- df = df[final_cols]
69
  df = df.round(decimals=2)
70
 
 
71
  return df
72
 
73
 
 
55
  if AutoEvalColumn.average.name in df.columns:
56
  df = df.sort_values(by=[AutoEvalColumn.average.name], ascending=False)
57
 
58
+ # Apenas arredondar os valores numéricos existentes
 
 
 
 
 
 
 
 
 
 
59
  df = df.round(decimals=2)
60
 
61
+ print(f"Colunas retornadas por get_leaderboard_df: {df.columns.tolist()}") # Adicionar log
62
  return df
63
 
64