switch to partition-wise sampling for better performance