Hey Tom, one issue with this PR -- you also need to add this to the SBT build. Try building the tests with SPARK_HADOOP_VERSION=2.0.5-alpha SPARK_YARN=true sbt/sbt test:compile