Spark: условно (`if`) для передачи значения параметра в функцию

Question

Я хочу передать значение параметра в функцию, но оцениваю значение другой переменной. Я пробовал это:

val myRDDResult = myRDD.mapPartitionsWithIndex( myFunction(index, partData, if (limit > partData.size) limit else limit - 1 ), preservesPartitioning = true)

Но не работает. И я также проверил это:

val myRDDResult = myRDD.mapPartitionsWithIndex( myFunction(index, partData, limit ), preservesPartitioning = true)

И это работает! Поэтому я думаю, что проблема заключается в использовании if внутри, но я не знаю, как я могу это сделать.