FIX: batchwise_shuffled_sequential returning wrong number of examples