debug

src/populate.py

@@ -68,12 +68,32 @@ def get_leaderboard_df_mib_subgraph(results_path: str, requests_path: str, cols:
 
 
 
+# def aggregate_methods(df: pd.DataFrame) -> pd.DataFrame:
+#     """Aggregates rows with the same base method name by taking the max value for each column"""
+#     df_copy = df.copy()
+    
+#     # Extract base method names (remove _2, _3, etc. suffixes)
+#     base_methods = [name.split('_')[0] if '_' in name and name.split('_')[-1].isdigit() 
+#                    else name for name in df_copy.index]
+#     df_copy.index = base_methods
+    
+#     # Convert scores to numeric values
+#     numeric_df = df_copy.select_dtypes(include=['float64', 'int64'])
+    
+#     # Group by base method name and take the max
+#     aggregated_df = numeric_df.groupby(level=0).max().round(3)
+    
+#     return aggregated_df
 def aggregate_methods(df: pd.DataFrame) -> pd.DataFrame:
     """Aggregates rows with the same base method name by taking the max value for each column"""
     df_copy = df.copy()
     
+    # Set Method as index if it isn't already
+    if 'Method' in df_copy.columns:
+        df_copy.set_index('Method', inplace=True)
+    
     # Extract base method names (remove _2, _3, etc. suffixes)
-    base_methods = [name.split('_')[0] if '_' in name and name.split('_')[-1].isdigit() 
+    base_methods = [name.split('_')[0] if '_' in str(name) and str(name).split('_')[-1].isdigit() 
                    else name for name in df_copy.index]
     df_copy.index = base_methods
     
@@ -83,6 +103,10 @@ def aggregate_methods(df: pd.DataFrame) -> pd.DataFrame:
     # Group by base method name and take the max
     aggregated_df = numeric_df.groupby(level=0).max().round(3)
     
+    # Reset index to get Method as a column
+    aggregated_df.reset_index(inplace=True)
+    aggregated_df.rename(columns={'index': 'Method'}, inplace=True)
+    
     return aggregated_df
 
 def create_intervention_averaged_df(df: pd.DataFrame) -> pd.DataFrame: