Closed Ig-dolci closed 7 months ago
Looks good. There are some comments regarding documentation and style in certain places.
Would also be good to a have a simple test for this in
pyadjoint
, not just in thefiredrake.adjoint
/firedrake_adjoint
framework.
It is hard to build a pyadjoint test since the checkpointing will not work with AdjFloat
. I do not understand deeply, but I think be no possible build Blocks
and its outputs and dependencies with test based in arrays.
Notes: Move the docstring from numpy-style to google-style.