Nathan Habib
commited on
Commit
•
eed1ccd
1
Parent(s):
a2790cb
fix list of finished models to include models waiting for new eval
Browse files- app.py +3 -3
- src/load_from_hub.py +1 -1
app.py
CHANGED
@@ -100,10 +100,10 @@ models = original_df["model_name_for_query"].tolist() # needed for model backlin
|
|
100 |
|
101 |
to_be_dumped = f"models = {repr(models)}\n"
|
102 |
|
103 |
-
with open("models_backlinks.py", "w") as f:
|
104 |
-
|
105 |
|
106 |
-
print(to_be_dumped)
|
107 |
|
108 |
leaderboard_df = original_df.copy()
|
109 |
(
|
|
|
100 |
|
101 |
to_be_dumped = f"models = {repr(models)}\n"
|
102 |
|
103 |
+
# with open("models_backlinks.py", "w") as f:
|
104 |
+
# f.write(to_be_dumped)
|
105 |
|
106 |
+
# print(to_be_dumped)
|
107 |
|
108 |
leaderboard_df = original_df.copy()
|
109 |
(
|
src/load_from_hub.py
CHANGED
@@ -128,7 +128,7 @@ def get_evaluation_queue_df(
|
|
128 |
|
129 |
pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
|
130 |
running_list = [e for e in all_evals if e["status"] == "RUNNING"]
|
131 |
-
finished_list = [e for e in all_evals if e["status"].startswith("FINISHED")]
|
132 |
df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
|
133 |
df_running = pd.DataFrame.from_records(running_list, columns=cols)
|
134 |
df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
|
|
|
128 |
|
129 |
pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
|
130 |
running_list = [e for e in all_evals if e["status"] == "RUNNING"]
|
131 |
+
finished_list = [e for e in all_evals if e["status"].startswith("FINISHED") or e["status"] == "PENDING_NEW_EVAL"]
|
132 |
df_pending = pd.DataFrame.from_records(pending_list, columns=cols)
|
133 |
df_running = pd.DataFrame.from_records(running_list, columns=cols)
|
134 |
df_finished = pd.DataFrame.from_records(finished_list, columns=cols)
|