Spaces:
Sleeping
Sleeping
- 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 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
|
53 |
-
|
54 |
-
|
55 |
|
56 |
-
|
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"]
|