Replace np.random.choice with utils.random.sample_without_replacement