Removed the TODO comment from PartitionPruningRDD.