Update src/get_results_for_task.py
Browse files
src/get_results_for_task.py
CHANGED
@@ -79,6 +79,8 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
|
|
79 |
model_hyperlink(link=link, model_name=model_name) if link else model_name
|
80 |
for link, model_name in zip(results_df["model_url"], results_df["Model Name"])
|
81 |
]
|
|
|
|
|
82 |
results_df["Resources"] = [_process_urls(urls) for urls in results_df["Resources"]]
|
83 |
results_df = results_df[get_columns_per_task(task_id)]
|
84 |
return results_df
|
|
|
79 |
model_hyperlink(link=link, model_name=model_name) if link else model_name
|
80 |
for link, model_name in zip(results_df["model_url"], results_df["Model Name"])
|
81 |
]
|
82 |
+
if task_id == 'project_code_completion':
|
83 |
+
results_df["Dataset"] = [_process_urls(urls) for urls in results_df["Dataset"]]
|
84 |
results_df["Resources"] = [_process_urls(urls) for urls in results_df["Resources"]]
|
85 |
results_df = results_df[get_columns_per_task(task_id)]
|
86 |
return results_df
|