Add dimension check to sample_discrete and sample_likelihoods