update populate
Browse files- src/populate.py +19 -1
src/populate.py
CHANGED
@@ -42,6 +42,24 @@ def get_leaderboard_df(eval_results_path, eval_requests_path, cols, benchmark_co
|
|
42 |
if 'average' in df.columns:
|
43 |
df = df.sort_values(by=['average'], ascending=False)
|
44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
45 |
return df
|
46 |
|
47 |
def get_evaluation_queue_df(eval_requests_path, eval_cols):
|
@@ -69,4 +87,4 @@ def get_evaluation_queue_df(eval_requests_path, eval_cols):
|
|
69 |
running_df = df[df['status'] == 'running']
|
70 |
pending_df = df[df['status'] == 'pending']
|
71 |
|
72 |
-
return finished_df, running_df, pending_df
|
|
|
42 |
if 'average' in df.columns:
|
43 |
df = df.sort_values(by=['average'], ascending=False)
|
44 |
|
45 |
+
# 🔹 Fix Column Data Types for Correct Display in Gradio
|
46 |
+
NUMERIC_COLUMNS = ["average", "likes", "params"] + [col for col in df.columns if col not in [
|
47 |
+
"model_name", "submitted_time", "base_model", "revision", "precision", "weight_type", "model_type", "license"
|
48 |
+
]]
|
49 |
+
|
50 |
+
for col in NUMERIC_COLUMNS:
|
51 |
+
if col in df.columns:
|
52 |
+
df[col] = pd.to_numeric(df[col], errors="coerce")
|
53 |
+
|
54 |
+
TEXT_COLUMNS = ["model_name", "submitted_time", "base_model", "revision", "precision", "weight_type", "model_type", "license"]
|
55 |
+
|
56 |
+
for col in TEXT_COLUMNS:
|
57 |
+
if col in df.columns:
|
58 |
+
df[col] = df[col].astype(str)
|
59 |
+
|
60 |
+
# Debugging: Print data types to confirm correctness
|
61 |
+
print("✅ LEADERBOARD_DF Updated dtypes:\n", df.dtypes)
|
62 |
+
|
63 |
return df
|
64 |
|
65 |
def get_evaluation_queue_df(eval_requests_path, eval_cols):
|
|
|
87 |
running_df = df[df['status'] == 'running']
|
88 |
pending_df = df[df['status'] == 'pending']
|
89 |
|
90 |
+
return finished_df, running_df, pending_df
|