This import should be moved up. See the style guide:

https://cwiki.apache.org/confluence/display/SPARK/Spark+Code+Style+Guide