Hmm... Sounds good. I should take care of it, especially when the data in each partition is little and the `miniBatchFraction` is low, a blank partition could occur.