MAINT: replace a component of the rao-teh sampler with a more generalized implementation