MAINT: reorganize the sampling code in preparation for computing conditional Holmes-Rubin-style expectations