fix: Use RDD partition index#1112

viirya merged 4 commits intoapache:mainapache/datafusion-comet:mainfrom

viirya:fix_partition_indexviirya/arrow-datafusion-comet:fix_partition_indexCopy head branch name to clipboard