Only creating a single partition per row group for indexed RDDs