Modified MappedSource.partition to actually partition the underlying
source.