From 4399b32e093d84d667f8ebd8843905b9501a0408 Mon Sep 17 00:00:00 2001 From: Iglesia Dolci Date: Tue, 10 Dec 2024 15:48:50 +0000 Subject: [PATCH] More docs --- firedrake/adjoint/ensemble_reduced_functional.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/firedrake/adjoint/ensemble_reduced_functional.py b/firedrake/adjoint/ensemble_reduced_functional.py index 28ed23c140..099831f6b6 100644 --- a/firedrake/adjoint/ensemble_reduced_functional.py +++ b/firedrake/adjoint/ensemble_reduced_functional.py @@ -84,13 +84,15 @@ def __init__(self, J, control, ensemble, scatter_control=True, scale=1.0, tape=None, eval_cb_pre=lambda *args: None, eval_cb_post=lambda *args: None, derivative_cb_pre=lambda controls: controls, - derivative_cb_post=lambda checkpoint, derivative_components, controls: derivative_components - ): + derivative_cb_post=lambda checkpoint, derivative_components, controls: derivative_components, + hessian_cb_pre=lambda *args: None, hessian_cb_post=lambda *args: None): super(EnsembleReducedFunctional, self).__init__( J, control, derivative_components=derivative_components, scale=scale, tape=tape, eval_cb_pre=eval_cb_pre, eval_cb_post=eval_cb_post, derivative_cb_pre=derivative_cb_pre, - derivative_cb_post=derivative_cb_post) + derivative_cb_post=derivative_cb_post, + hessian_cb_pre=hessian_cb_pre, hessian_cb_post=hessian_cb_post) + self.ensemble = ensemble self.scatter_control = scatter_control self.gather_functional = gather_functional