OK, thanks Ricardo! That shifts my attention a little bit. Perhaps this whole issue should be about identifying the source of that warning first, rather than verifying that minibatch is rotating indices.
Do you have any insight into what causes the warning? I couldn’t find any previous issues with exactly this warning