Pyspark dataset, 10 is required, with 3