Nathan Habib commited on
Commit
eed1ccd
1 Parent(s): a2790cb

fix list of finished models to include models waiting for new eval

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. 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
- f.write(to_be_dumped)
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)