[SPARK-3170][CORE]: Bug Fix in Storage UI

current compeleted stage only need to remove its own partitions that are no longer cached. Currently, "Storage" in Spark UI may lost some rdds which are cached actually.
@uncleGen Tests are failing because this line is too long. Can you wrap this into 2 lines?