style fix

apache · Jul 8, 2014 · 9884a9f · 9884a9f
1 parent bbfb8c9
commit 9884a9f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/core/src/main/scala/org/apache/spark/rdd/PairRDDFunctions.scala b/core/src/main/scala/org/apache/spark/rdd/PairRDDFunctions.scala
@@ -216,9 +216,9 @@ class PairRDDFunctions[K, V](self: RDD[(K, V)])
 
     val samplingFunc = if (withReplacement) {
       val counts = if (exact) Some(this.countByKey()) else None
-        StratifiedSampler.getPoissonSamplingFunction(self, fractions, exact, counts, seed)
+      StratifiedSampler.getPoissonSamplingFunction(self, fractions, exact, counts, seed)
     } else {
-        StratifiedSampler.getBernoulliSamplingFunction(self, fractions, exact, seed)
+      StratifiedSampler.getBernoulliSamplingFunction(self, fractions, exact, seed)
     }
     self.mapPartitionsWithIndex(samplingFunc, preservesPartitioning=true)
   }

diff --git a/core/src/main/scala/org/apache/spark/util/random/StratifiedSampler.scala b/core/src/main/scala/org/apache/spark/util/random/StratifiedSampler.scala
@@ -155,8 +155,7 @@ private[spark] object StratifiedSampler extends Logging {
    * to be included in the sample.
    */
   def computeThresholdByKey[K](finalResult: Map[K, Stratum],
-      fractions: Map[K, Double]):
-    Map[K, Double] = {
+      fractions: Map[K, Double]): Map[K, Double] = {
     val thresholdByKey = new HashMap[K, Double]()
     for ((key, stratum) <- finalResult) {
       val s = math.ceil(stratum.numItems * fractions(key)).toLong