By the way, I would rename random_idxs to `subset` to be coherent.