caulsal-track debug

src/populate.py

@@ -39,9 +39,7 @@ def get_leaderboard_df_mib_subgraph(results_path: str, requests_path: str, cols:
     """Creates a dataframe from all the MIB experiment results"""
     # print(f"results_path is {results_path}, requests_path is {requests_path}")
     raw_data = get_raw_eval_results_mib_subgraph(results_path, requests_path)
-    # print(f"raw_data is {raw_data}")
-    
-    # Convert each result to dict format
+
     all_data_json = [v.to_dict(metric_type=metric_type) for v in raw_data]
     # print(f"all_data_json is {pd.DataFrame.from_records(all_data_json)}")
 
@@ -127,16 +125,16 @@ def create_intervention_averaged_df(df: pd.DataFrame) -> pd.DataFrame:
 def get_leaderboard_df_mib_causalgraph(results_path: str, requests_path: str) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
     # print(f"results_path is {results_path}, requests_path is {requests_path}")
 
-    raw_detailed_df, raw_aggregated_df, raw_intervention_averaged_df = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
+    detailed_df, aggregated_df, intervention_averaged_df = get_raw_eval_results_mib_causalgraph(results_path, requests_path)
 
-    all_data_json = [v.to_dict() for v in raw_detailed_df]
-    detailed_df = pd.DataFrame.from_records(all_data_json)
+    # all_data_json = [v.to_dict() for v in raw_detailed_df]
+    # detailed_df = pd.DataFrame.from_records(all_data_json)
 
-    all_data_json = [v.to_dict() for v in raw_aggregated_df]
-    aggregated_df = pd.DataFrame.from_records(all_data_json)
+    # all_data_json = [v.to_dict() for v in raw_aggregated_df]
+    # aggregated_df = pd.DataFrame.from_records(all_data_json)
 
-    all_data_json = [v.to_dict() for v in raw_intervention_averaged_df]
-    intervention_averaged_df = pd.DataFrame.from_records(all_data_json)
+    # all_data_json = [v.to_dict() for v in raw_intervention_averaged_df]
+    # intervention_averaged_df = pd.DataFrame.from_records(all_data_json)
     
     # # Rename columns to match schema
     # column_mapping = {}