The docs of partition say that
stratify=nothing: if a vector is specified, the partition will match the stratification of the given vector. In that case, shuffle cannot be false`.
However,
shuffle is internally set to true and a shuffle performed if an RNG is supplied, even if shuffle=false (i.e. the docs should probably also say don't supply an RNG either?)
- I'm fairly confident, that the
stratify vector is actually correctly indexed into (here) even if shuffle=true. (Ideally, someone with more code familiarity would quickly look over this, too, though—me being fairly confident may not be enough for the standards of this repo.)
If I'm mistaken, then sorry for the noise.
If not, then it may make sense to update the docs (in either case)?