fixed parallel sampling when n_samples % n_chains != 0