SPARK-1939 Refactor takeSample method in RDD to use ScaSRS