Is it possible to set the final state of the flow to the state of one of the intermediate tasks?

Yeah use:

with Flow(...) as flow:
    a = ...

flow.set_reference_tasks([a])

this was for 1.0, if you need it for 2.0, check this

Amazing, thank you guys!