Fix some shuffle docs
The docs for this PR live [here](https://moon-ci-docs.huggingface.co/docs/datasets/pr_3885). All of your documentation changes will be reflected on that endpoint.
Following #3842 some docs were still outdated (with `buffer_size` as the first argument)