Update PySpark to be less OOM-prone as well
Update comment to better reflect what we're doing
Quadruple instead of double for a minor speedup

Speedup occurs when calling take() on an RDD with most partitions empty