Add dataset field to code completion task

src/leaderboard_formatting.py

@@ -20,6 +20,7 @@ COLUMNS_PRETTY = {
     "EM non_informative": "EM non-informative",
     "EM random": "EM random",
     "EM all": "EM all",    
+    "dataset": "Dataset",
 }
 
 
@@ -47,16 +48,19 @@ METRICS_PER_TASK = {
 
 SORT_COLUMN_PER_TASK = {
     "commit_message_generation": "ROUGE-1",
-    "project_code_completion": "EM infile"
+    "project_code_completion": "Dataset"
 }
 
 
 def get_columns_per_task(task_id: str) -> List[str]:
     metrics_per_task = METRICS_PER_TASK[task_id]
-
+    if task_id == 'project_code_completion':
+        return ["Model Name", "Availability", "Context Size", "Dataset"] + metrics_per_task + ["Submitted By", "Resources"]
     return ["Model Name", "Availability", "Context Size"] + metrics_per_task + ["Submitted By", "Resources"]
 
 
 def get_types_per_task(task_id: str) -> List[str]:
     metrics_per_task = METRICS_PER_TASK.get(task_id, (0, 0, 0, 0, 0))
+    if task_id == 'project_code_completion':
+        return ["html", "markdown", "markdown", "html"] + ["number" for _ in metrics_per_task] + ["markdown", "html"]
     return ["html", "markdown", "markdown"] + ["number" for _ in metrics_per_task] + ["markdown", "html"]