Why doesn't this just call `rdd.unpersist()`? It seems we have duplicated code paths here.