add titles option to plots_2d

xgarrido · Oct 12, 2023 · 8e96c22 · 8e96c22
1 parent 7a3e0a6
commit 8e96c22
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cobaya_utilities/plots.py b/cobaya_utilities/plots.py
@@ -191,8 +191,9 @@ def plots_2d(*args, **kwargs):
     g = get_subplot_plotter(settings=get_default_settings(), **plotter_kwargs)
     g.plots_2d(*args, **kwargs)
 
-    if kwargs.get("despine", True):
-        despine(g)
+    if titles := kwargs.get("titles"):
+        for title, ax in zip(titles, g.fig.axes):
+            ax.set_title(title)
 
     return g