Commit
7a4d3ec
·
verified ·
1 Parent(s): f177bfb

update populate

Browse files
Files changed (1) hide show
  1. 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