The partitionByIndex method of MiniBatchable Datastore is incorrectly defined. The LHS argument returned by partitionByIndex must be of the same class as the input datastore