Equivalent for assemble in dolfinx

[nate]

See, for example, @dokken’s excellent dolfinx tutorial for guides on updating your code.