DOC shape=(model.size_psi,), not (model.psi,)