The partitionByIndex method of MiniBatchable Datastore is incorrectly defined. The datastore returned by partitionByIndex must always contain the requested number of indices