Fix formatting
Browse files- app.py +1 -1
- src/get_results_for_task.py +8 -1
app.py
CHANGED
@@ -25,7 +25,7 @@ submission_uploader = SubmissionUploader(os.environ["DATASET_ID"])
|
|
25 |
|
26 |
def restart_space():
|
27 |
HfApi(token=os.environ["HF_TOKEN"]).restart_space(
|
28 |
-
repo_id="JetBrains-Research/long-code-arena"
|
29 |
)
|
30 |
|
31 |
|
|
|
25 |
|
26 |
def restart_space():
|
27 |
HfApi(token=os.environ["HF_TOKEN"]).restart_space(
|
28 |
+
repo_id="JetBrains-Research/long-code-arena", token=os.environ["HF_TOKEN"]
|
29 |
)
|
30 |
|
31 |
|
src/get_results_for_task.py
CHANGED
@@ -57,7 +57,14 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
|
|
57 |
)
|
58 |
|
59 |
for metric_column in METRICS_PER_TASK[task_id]:
|
60 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
|
62 |
results_df = results_df[get_columns_per_task(task_id)]
|
63 |
return results_df
|
|
|
57 |
)
|
58 |
|
59 |
for metric_column in METRICS_PER_TASK[task_id]:
|
60 |
+
if "BERTScore" in metric_column:
|
61 |
+
results_df[metric_column] = results_df[metric_column].map(
|
62 |
+
lambda x: f"{x:.5f}"
|
63 |
+
)
|
64 |
+
else:
|
65 |
+
results_df[metric_column] = results_df[metric_column].map(
|
66 |
+
lambda x: f"{x:.2f}"
|
67 |
+
)
|
68 |
|
69 |
results_df = results_df[get_columns_per_task(task_id)]
|
70 |
return results_df
|