Hi,
I am learning the dataset API 2. The shuffle API states that it rearranges the values of a column but from my experimentations, it shuffles the rows.
The code documentation 2 is more clear and states that the rows are shuffled.
To achieve column shuffle I used the map functionality (batch=True) and created the following mapper function:
def _shuffle_question_column_batch(examples):
    questions = examples["question"]
    Random(42).shuffle(questions)
    examples["question"] = questions
    return examples

I am wondering whether the shuffle API is capable of rearranging values of a single column or a better way exists.
Please advice.