Update src/vis_utils.py

src/vis_utils.py

@@ -230,7 +230,7 @@ def plot_affinity_results(method_names, metric, affinity_path="/tmp/affinity_res
     
     # Gather columns related to the specified metric and validate
     metric_columns = [col for col in df.columns if col.startswith(f"{metric}_")]
-    df = df[['Method'] + metric_columns]
+    df = df[['Method'] + metric_columns].set_index('Method')
 
     df = df.fillna(0)