Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Added failed row
Browse files- src/populate.py +4 -1
src/populate.py
CHANGED
@@ -53,7 +53,10 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
|
|
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)
|
57 |
df_running = pd.DataFrame.from_records(running_list, columns=cols)
|
58 |
df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
|
59 |
-
|
|
|
|
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 |
+
failed_list = [e for e in all_evals if e["status"] == "FAILED"]
|
57 |
+
|
58 |
df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
|
59 |
df_running = pd.DataFrame.from_records(running_list, columns=cols)
|
60 |
df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
|
61 |
+
df_failed = pd.DataFrame.from_records(failed_list, columns=cols)
|
62 |
+
return df_finished[cols], df_running[cols], df_pending[cols], df_failed[cols]
|