jenyag commited on
Commit
fcf16bd
1 Parent(s): f2f6f8f

Update src/get_results_for_task.py

Browse files
Files changed (1) hide show
  1. src/get_results_for_task.py +2 -0
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