MAINT: generalize rao-teh sampler but no extra tests have been added