lixuejing
commited on
Commit
·
5735862
1
Parent(s):
4b381d6
add log
Browse files- src/leaderboard/read_evals.py +1 -0
- src/tools/plots.py +2 -1
src/leaderboard/read_evals.py
CHANGED
|
@@ -161,6 +161,7 @@ class EvalResult:
|
|
| 161 |
|
| 162 |
def get_request_file_for_model(requests_path, model_name, precision):
|
| 163 |
"""Selects the correct request file for a given model. Only keeps runs tagged as FINISHED"""
|
|
|
|
| 164 |
request_files = os.path.join(
|
| 165 |
requests_path,
|
| 166 |
f"{model_name}_eval_request_*.json",
|
|
|
|
| 161 |
|
| 162 |
def get_request_file_for_model(requests_path, model_name, precision):
|
| 163 |
"""Selects the correct request file for a given model. Only keeps runs tagged as FINISHED"""
|
| 164 |
+
print("get_request_file_for_model", requests_path, model_name)
|
| 165 |
request_files = os.path.join(
|
| 166 |
requests_path,
|
| 167 |
f"{model_name}_eval_request_*.json",
|
src/tools/plots.py
CHANGED
|
@@ -19,7 +19,8 @@ def create_scores_df(raw_data: list[EvalResult]) -> pd.DataFrame:
|
|
| 19 |
"""
|
| 20 |
# Step 1: Ensure 'date' is in datetime format and sort the DataFrame by it
|
| 21 |
results_df = pd.DataFrame(raw_data)
|
| 22 |
-
|
|
|
|
| 23 |
results_df.sort_values(by="date", inplace=True)
|
| 24 |
|
| 25 |
# Step 2: Initialize the scores dictionary
|
|
|
|
| 19 |
"""
|
| 20 |
# Step 1: Ensure 'date' is in datetime format and sort the DataFrame by it
|
| 21 |
results_df = pd.DataFrame(raw_data)
|
| 22 |
+
print("results_df", results_df)
|
| 23 |
+
results_df["date"] = pd.to_datetime(results_df["date"], format="mixed", utc=True)
|
| 24 |
results_df.sort_values(by="date", inplace=True)
|
| 25 |
|
| 26 |
# Step 2: Initialize the scores dictionary
|