COSMIT handle multi-output more explicitly

May also fix untested single-sample predictions