Spaces:
Running
Running
Update src/populate.py
Browse files- 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 |
-
#
|
| 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 |
|