Change PySpark RDD.take() to not call iterator().