Update src/vis_utils.py

src/vis_utils.py

@@ -150,7 +150,7 @@ def plot_function_results(aspect, metric, method_names, function_path="/tmp/func
     # Create clustermap
     g = sns.clustermap(df, annot=True, cmap="YlGnBu", row_cluster=False, col_cluster=False, figsize=(15, 15))
 
-    title = f"{long_form_mapping[aspect.capitalize()]} Results for {metric.capitalize()}"
+    title = f"{long_form_mapping[aspect.upper()]} Results for {metric.capitalize()}"
     g.fig.suptitle(title, x=0.5, y=1.02, fontsize=16, ha='center')  # Center the title above the plot
 
     # Get heatmap axis and customize labels