karimouda commited on
Commit
92e74cb
·
1 Parent(s): ead29cd
Files changed (1) hide show
  1. src/populate.py +2 -0
src/populate.py CHANGED
@@ -49,8 +49,10 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
49
  data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
50
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
51
  all_evals.append(data)
 
52
 
53
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
 
54
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
55
  finished_list = [e for e in all_evals if e["status"].startswith("FINISHED") or e["status"] == "PENDING_NEW_EVAL"]
56
  df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
 
49
  data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
50
  data[EvalQueueColumn.revision.name] = data.get("revision", "main")
51
  all_evals.append(data)
52
+ print(all_evals)
53
 
54
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
55
+ print(pending_list)
56
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
57
  finished_list = [e for e in all_evals if e["status"].startswith("FINISHED") or e["status"] == "PENDING_NEW_EVAL"]
58
  df_pending = pd.DataFrame.from_records(pending_list, columns=cols)