Fix plotting bug

Author: stefanradev93

.pre-commit-config.yaml

@@ -14,7 +14,7 @@
 repos:
 - repo: https://github.com/astral-sh/ruff-pre-commit
   # Ruff version.
-  rev: v0.9.6
+  rev: v0.9.10
   hooks:
     # Run the linter.
     - id: ruff

bayesflow/diagnostics/plots/z_score_contraction.py

@@ -121,7 +121,7 @@ def z_score_contraction(
         ax.scatter(contraction[:, i], z_score[:, i], color=color, alpha=0.5)
         ax.set_xlim([-0.05, 1.05])
 
-    prettify_subplots(plot_data["axes"], tick_fontsize)
+    prettify_subplots(plot_data["axes"], num_subplots=plot_data["num_variables"], tick_fontsize=tick_fontsize)
 
     # Add labels, titles, and set font sizes
     add_titles_and_labels(

bayesflow/utils/plot_utils.py

@@ -203,8 +203,8 @@ def add_y_labels(axes: np.ndarray, num_row: int = None, ylabel: Sequence[str] |
 
 
 def add_titles(axes: np.ndarray, title: Sequence[str] | str = None, title_fontsize: int = None):
-    for i, ax in enumerate(axes.flat):
-        ax.set_title(title[i], fontsize=title_fontsize)
+    for t, ax in zip(title, axes.flat):
+        ax.set_title(t, fontsize=title_fontsize)
 
 
 def add_titles_and_labels(