t0-0 commited on
Commit
6e6645b
1 Parent(s): 28dbc1f

Update src/populate.py

Browse files
Files changed (1) hide show
  1. src/populate.py +3 -1
src/populate.py CHANGED
@@ -61,7 +61,9 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
61
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
62
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
63
  finished_list = [e for e in all_evals if e["status"].startswith("FINISHED") or e["status"] == "PENDING_NEW_EVAL"]
 
64
  df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
65
  df_running = pd.DataFrame.from_records(running_list, columns=cols)
66
  df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
67
- return df_finished[cols], df_running[cols], df_pending[cols]
 
 
61
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
62
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
63
  finished_list = [e for e in all_evals if e["status"].startswith("FINISHED") or e["status"] == "PENDING_NEW_EVAL"]
64
+ failed_list = [e for e in all_evals if e["status"] == "FAILED"]
65
  df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
66
  df_running = pd.DataFrame.from_records(running_list, columns=cols)
67
  df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
68
+ df_failed = pd.DataFrame.from_records(failed_list, columns=cols)
69
+ return df_finished[cols], df_running[cols], df_pending[cols], df_failed[cols]