PartitionPruningRDD is using index from parent