Java Multi Line Lambda Functions In Pyspark