fn split_combined_batch_by_region( combined_batch: &RecordBatch, partition_rule: &dyn PartitionRule, ) -> Result<HashMap<u32, RegionMask>>