Fix formatting

app.py

@@ -25,7 +25,7 @@ submission_uploader = SubmissionUploader(os.environ["DATASET_ID"])
 
 def restart_space():
     HfApi(token=os.environ["HF_TOKEN"]).restart_space(
-        repo_id="JetBrains-Research/long-code-arena"
+        repo_id="JetBrains-Research/long-code-arena", token=os.environ["HF_TOKEN"]
     )
 
 

src/get_results_for_task.py

@@ -57,7 +57,14 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
     )
 
     for metric_column in METRICS_PER_TASK[task_id]:
-        results_df[metric_column] = results_df[metric_column].map(lambda x: f"{x:.2f}")
+        if "BERTScore" in metric_column:
+            results_df[metric_column] = results_df[metric_column].map(
+                lambda x: f"{x:.5f}"
+            )
+        else:
+            results_df[metric_column] = results_df[metric_column].map(
+                lambda x: f"{x:.2f}"
+            )
 
     results_df = results_df[get_columns_per_task(task_id)]
     return results_df