From 2c0af8d1078b27899336baa10f67ddffaa31bee9 Mon Sep 17 00:00:00 2001
From: Daniel Weindl <daniel.weindl@uni-bonn.de>
Date: Sat, 1 Nov 2025 11:39:12 +0100
Subject: [PATCH] Include id in plot titles

* When plotting trajectories, include `ReturnData.id` in the title if set.
* Set `amici.plotting.__all__`
---
 python/sdist/amici/plotting.py | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/python/sdist/amici/plotting.py b/python/sdist/amici/plotting.py
index f3388c346d..987b890890 100644
--- a/python/sdist/amici/plotting.py
+++ b/python/sdist/amici/plotting.py
@@ -17,6 +17,13 @@
 from . import Model, ReturnDataView
 from .numpy import StrOrExpr, evaluate
 
+__all__ = [
+    "plot_state_trajectories",
+    "plot_observable_trajectories",
+    "plot_jacobian",
+    "plot_expressions",
+]
+
 
 def plot_state_trajectories(
     rdata: ReturnDataView,
@@ -73,7 +80,11 @@ def plot_state_trajectories(
     ax.set_xlabel("$t$")
     ax.set_ylabel("$x(t)$")
     ax.legend()
-    ax.set_title("State trajectories")
+
+    title = "State trajectories"
+    if rdata.id:
+        title += f" – {rdata.id}"
+    ax.set_title(title)
 
 
 def plot_observable_trajectories(
@@ -158,9 +169,13 @@ def plot_observable_trajectories(
 
     ax.set_xlabel("$t$")
     ax.set_ylabel("$y(t)$")
-    ax.set_title("Observable trajectories")
     ax.legend()
 
+    title = "Observable trajectories"
+    if rdata.id:
+        title += f" – {rdata.id}"
+    ax.set_title(title)
+
 
 def plot_jacobian(rdata: ReturnDataView):
     """Plot Jacobian as heatmap."""