SPARK-1019: pyspark RDD take() throws an NPE
Update script/setup to use assembly as new spark versions require