[SPARK-3625] In some cases, the RDD.checkpoint does not work

There is no explanation of what this change is trying to accomplish. What do you believe this fixes? I would not use "1" and "2" to represent some kind of state without explanation.