hSterz commited on
Commit
574f1e1
·
1 Parent(s): a617daf
Files changed (1) hide show
  1. src/populate.py +7 -7
src/populate.py CHANGED
@@ -45,15 +45,15 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
45
  print(entries)
46
  all_evals = []
47
 
48
- # for entry in entries:
49
- # file_path = os.path.join(save_path, entry, "eval_request.json")
50
- # with open(file_path) as fp:
51
- # data = json.load(fp)
52
 
53
- # data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
54
- # data[EvalQueueColumn.revision.name] = data.get("revision", "main")
55
 
56
- # all_evals.append(data)
57
 
58
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
59
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
 
45
  print(entries)
46
  all_evals = []
47
 
48
+ for entry in entries:
49
+ file_path = os.path.join(save_path, entry, "eval_request.json")
50
+ with open(file_path) as fp:
51
+ data = json.load(fp)
52
 
53
+ data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
54
+ data[EvalQueueColumn.revision.name] = data.get("revision", "main")
55
 
56
+ all_evals.append(data)
57
 
58
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
59
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]