galtimur commited on
Commit
f1b1d8d
·
1 Parent(s): fd11dec

Renamed the column

Browse files
Files changed (1) hide show
  1. src/get_results_for_task.py +2 -0
src/get_results_for_task.py CHANGED
@@ -96,6 +96,8 @@ def _get_results_dataset(task_id: str) -> pd.DataFrame:
96
  results_df["Dataset"] = [_process_urls(urls) for urls in results_df["Dataset"]]
97
  results_df["Resources"] = [_process_urls(urls) for urls in results_df["Resources"]]
98
  results_df = results_df[get_columns_per_task(task_id)]
 
 
99
  return results_df
100
 
101
 
 
96
  results_df["Dataset"] = [_process_urls(urls) for urls in results_df["Dataset"]]
97
  results_df["Resources"] = [_process_urls(urls) for urls in results_df["Resources"]]
98
  results_df = results_df[get_columns_per_task(task_id)]
99
+ if task_id == 'ci_builds_repair':
100
+ results_df = results_df.rename(columns={"Context Size": "Context"})
101
  return results_df
102
 
103