PartitionPruningRDD is using index from parent(review changes)