Spaces:
Sleeping
Sleeping
- 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 |
-
|
54 |
-
|
55 |
-
|
|
|
56 |
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
|
62 |
-
|
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"]
|