We should never set the partitions to None.
Ensure the data hasn't changed in edge cases.
Wait longer for partition acquisition for pypy.