From 4f3c745feaefab353d1cbd66caef6d8d0cfb5433 Mon Sep 17 00:00:00 2001 From: Mattijs Borst Date: Fri, 18 Sep 2026 11:22:37 +0200 Subject: [PATCH 1/2] Expose neglect_steady in transient Model.headalongline Forwards the existing head() neglect_steady flag through headalongline so transient-only heads along a line can be computed once a steady model is linked via Model(steady=...). Co-Authored-By: Claude Sonnet 5 --- timflow/transient/model.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/timflow/transient/model.py b/timflow/transient/model.py index 263c27d8..828ce4e6 100644 --- a/timflow/transient/model.py +++ b/timflow/transient/model.py @@ -671,7 +671,7 @@ def headinside(self, elabel, t): def strength(self, elabel, t): return self.elementdict[elabel].strength(t - self.tstart) - def headalongline(self, x, y, t, layers=None): + def headalongline(self, x, y, t, layers=None, neglect_steady=False): """Head along line or curve. Parameters @@ -684,6 +684,9 @@ def headalongline(self, x, y, t, layers=None): times for which grid is returned layers : integer, list or array, optional layers for which grid is returned + neglect_steady : bool, optional + if `True`, the steady-state contribution from `self.steady` (if set) + is not added to the result, by default `False` Returns ------- @@ -701,7 +704,9 @@ def headalongline(self, x, y, t, layers=None): t = np.atleast_1d(t) h = np.zeros((Nlayers, len(t), nx)) for i in range(nx): - h[:, :, i] = self.head(xg[i], yg[i], t, layers) + h[:, :, i] = self.head( + xg[i], yg[i], t, layers, neglect_steady=neglect_steady + ) return h def disvecalongline(self, x, y, t, layers=None): From c1b964586d4e64a402329c340992f2a17272c084 Mon Sep 17 00:00:00 2001 From: Mattijs Borst Date: Fri, 18 Sep 2026 11:27:02 +0200 Subject: [PATCH 2/2] ruff --- timflow/transient/model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/timflow/transient/model.py b/timflow/transient/model.py index 828ce4e6..685270e5 100644 --- a/timflow/transient/model.py +++ b/timflow/transient/model.py @@ -704,9 +704,7 @@ def headalongline(self, x, y, t, layers=None, neglect_steady=False): t = np.atleast_1d(t) h = np.zeros((Nlayers, len(t), nx)) for i in range(nx): - h[:, :, i] = self.head( - xg[i], yg[i], t, layers, neglect_steady=neglect_steady - ) + h[:, :, i] = self.head(xg[i], yg[i], t, layers, neglect_steady=neglect_steady) return h def disvecalongline(self, x, y, t, layers=None):