From 68cb8e792ecd9cfe2a6e4445f7bea790f77d8dbb Mon Sep 17 00:00:00 2001 From: JHopeCollins Date: Tue, 7 Jan 2025 16:41:40 +0000 Subject: [PATCH] add markings argument to FunctionMergeBlock.evaluate_tlm --- firedrake/adjoint_utils/blocks/function.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/firedrake/adjoint_utils/blocks/function.py b/firedrake/adjoint_utils/blocks/function.py index e31a0c4567..dcb02da108 100644 --- a/firedrake/adjoint_utils/blocks/function.py +++ b/firedrake/adjoint_utils/blocks/function.py @@ -242,11 +242,13 @@ def evaluate_adj_component(self, inputs, adj_inputs, block_variable, idx, else: return adj_inputs[0] - def evaluate_tlm(self): + def evaluate_tlm(self, markings=False): tlm_input = self.get_dependencies()[0].tlm_value if tlm_input is None: return output = self.get_outputs()[0] + if markings and not output.marked_in_path: + return fs = output.output.function_space() f = type(output.output)(fs) output.add_tlm_output(