Update src/get_results_for_task.py

src/get_results_for_task.py

@@ -63,7 +63,7 @@ def _process_urls(raw_urls: str) -> str:
 def _extract_dataset_name(raw_urls: str) -> str:
     if not raw_urls:
         return raw_urls
-    names = [re.search(MD_LINK_PATTERN, url.strip()).group(1) for url in raw_urls.split(",")]
+    names = [re.search(MD_LINK_PATTERN, url.strip()).group(1) + ' context' for url in raw_urls.split(",")]
     return ", ".join(names)
     
 
@@ -88,7 +88,7 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
         for link, model_name in zip(results_df["model_url"], results_df["Model Name"])
     ]
     if task_id == 'project_code_completion':
-        results_df["Dataset_Names"] = [_extract_dataset_name(urls) for urls in results_df["Dataset"]]
+        results_df["Dataset Name"] = [_extract_dataset_name(urls) for urls in results_df["Dataset"]]
         results_df["Dataset"] = [_process_urls(urls) for urls in results_df["Dataset"]]
     results_df["Resources"] = [_process_urls(urls) for urls in results_df["Resources"]]
     results_df = results_df[get_columns_per_task(task_id)]