hSterz commited on
Commit
270b1f6
·
1 Parent(s): 53d2ed8
Files changed (1) hide show
  1. src/populate.py +9 -8
src/populate.py CHANGED
@@ -50,16 +50,17 @@ def get_evaluation_queue_df(save_path: str, cols: list) -> list[pd.DataFrame]:
50
  all_evals = []
51
 
52
  for entry in entries:
53
- file_path = os.path.join(save_path, entry, "eval_request.json")
54
- with open(file_path) as fp:
55
- data = json.load(fp)
 
56
 
57
- data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
58
- data[EvalQueueColumn.revision.name] = data.get("revision", "main")
59
- data[EvalQueueColumn.output_format.name] = data.get("output_format")
60
- data[EvalQueueColumn.dataset_version.name] = file_path.split("/")[-3]
61
 
62
- all_evals.append(data)
63
 
64
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
65
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]
 
50
  all_evals = []
51
 
52
  for entry in entries:
53
+ if not ".txt" in entry:
54
+ file_path = os.path.join(save_path, entry, "eval_request.json")
55
+ with open(file_path) as fp:
56
+ data = json.load(fp)
57
 
58
+ data[EvalQueueColumn.model.name] = make_clickable_model(data["model"])
59
+ data[EvalQueueColumn.revision.name] = data.get("revision", "main")
60
+ data[EvalQueueColumn.output_format.name] = data.get("output_format")
61
+ data[EvalQueueColumn.dataset_version.name] = file_path.split("/")[-3]
62
 
63
+ all_evals.append(data)
64
 
65
  pending_list = [e for e in all_evals if e["status"] in ["PENDING", "RERUN"]]
66
  running_list = [e for e in all_evals if e["status"] == "RUNNING"]