jasonshaoshun commited on
Commit
5f51841
·
1 Parent(s): 4d9550b

caulsal-track debug

Browse files
Files changed (1) hide show
  1. src/populate.py +4 -4
src/populate.py CHANGED
@@ -127,11 +127,11 @@ def create_intervention_averaged_df(df: pd.DataFrame) -> pd.DataFrame:
127
  def get_leaderboard_df_mib_causalgraph(results_path: str, requests_path: str) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
128
  # print(f"results_path is {results_path}, requests_path is {requests_path}")
129
 
130
- # raw_data = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
131
- # all_data_json = [v.to_dict() for v in raw_data]
132
- # detailed_df = pd.DataFrame.from_records(all_data_json)
133
 
134
- detailed_df = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
135
 
136
  # Print the actual columns for debugging
137
  print("Original columns:", detailed_df.columns.tolist())
 
127
  def get_leaderboard_df_mib_causalgraph(results_path: str, requests_path: str) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
128
  # print(f"results_path is {results_path}, requests_path is {requests_path}")
129
 
130
+ raw_data = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
131
+ all_data_json = [v.to_dict() for v in raw_data]
132
+ detailed_df = pd.DataFrame.from_records(all_data_json)
133
 
134
+ # detailed_df = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
135
 
136
  # Print the actual columns for debugging
137
  print("Original columns:", detailed_df.columns.tolist())