From ef0a86e41e9b390e6c0d60a6ed2105dbc54431f7 Mon Sep 17 00:00:00 2001 From: Josh Rosen Date: Wed, 13 May 2015 13:50:11 -0700 Subject: [PATCH] Fix scalastyle errors --- .../apache/spark/shuffle/unsafe/UnsafeShuffleSuite.scala | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/core/src/test/scala/org/apache/spark/shuffle/unsafe/UnsafeShuffleSuite.scala b/core/src/test/scala/org/apache/spark/shuffle/unsafe/UnsafeShuffleSuite.scala index 64569f1c60927..6351539e91e97 100644 --- a/core/src/test/scala/org/apache/spark/shuffle/unsafe/UnsafeShuffleSuite.scala +++ b/core/src/test/scala/org/apache/spark/shuffle/unsafe/UnsafeShuffleSuite.scala @@ -17,6 +17,8 @@ package org.apache.spark.shuffle.unsafe +import java.io.File + import scala.collection.JavaConverters._ import org.apache.commons.io.FileUtils @@ -51,7 +53,7 @@ class UnsafeShuffleSuite extends ShuffleSuite with BeforeAndAfterAll { .setSerializer(new KryoSerializer(myConf)) val shuffleDep = shuffledRdd.dependencies.head.asInstanceOf[ShuffleDependency[_, _, _]] assert(UnsafeShuffleManager.canUseUnsafeShuffle(shuffleDep)) - def getAllFiles = + def getAllFiles: Set[File] = FileUtils.listFiles(tmpDir, TrueFileFilter.INSTANCE, TrueFileFilter.INSTANCE).asScala.toSet val filesBeforeShuffle = getAllFiles // Force the shuffle to be performed @@ -82,7 +84,7 @@ class UnsafeShuffleSuite extends ShuffleSuite with BeforeAndAfterAll { .setSerializer(new JavaSerializer(myConf)) val shuffleDep = shuffledRdd.dependencies.head.asInstanceOf[ShuffleDependency[_, _, _]] assert(!UnsafeShuffleManager.canUseUnsafeShuffle(shuffleDep)) - def getAllFiles = + def getAllFiles: Set[File] = FileUtils.listFiles(tmpDir, TrueFileFilter.INSTANCE, TrueFileFilter.INSTANCE).asScala.toSet val filesBeforeShuffle = getAllFiles // Force the shuffle to be performed