Intellij Scala Spark transformation auto-completion?

Hello,

 Auto completion are very week when it comes to do Spark in Scala. Especially when implementing Spark transformations.

Is there a way to improve it to reach at least the same level as Java Spark ?

Thx!