Changed this now  -- `collectPartitions` takes an array of partitionIds and returns an array of Lists. The python `take` implementation still uses a single partition right now, but i have a TODO to match Scala that can be done in a separate PR.

cc @JoshRosen for the Python RDD changes.