From ce0e30c50d7b55c1aa598a0d1b49e2e9beff94a9 Mon Sep 17 00:00:00 2001
From: martinzapletal <zapletal-martin@email.cz>
Date: Sun, 18 Jan 2015 00:47:19 +0000
Subject: [PATCH] SPARK-3278 readability refactoring

---
 .../spark/mllib/regression/IsotonicRegression.scala   | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/regression/IsotonicRegression.scala b/mllib/src/main/scala/org/apache/spark/mllib/regression/IsotonicRegression.scala
index 376723310c74f..90c1f5a7347bc 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/regression/IsotonicRegression.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/regression/IsotonicRegression.scala
@@ -185,12 +185,11 @@ class PoolAdjacentViolators private [mllib]
       testData: RDD[(Double, Double, Double)],
       isotonic: Boolean): Seq[(Double, Double, Double)] = {
 
-    poolAdjacentViolators(
-      testData
-        .sortBy(_._2)
-        .cache()
-        .mapPartitions(it => poolAdjacentViolators(it.toArray, isotonic).toIterator)
-        .collect(), isotonic)
+    val parallelStepResult = testData
+      .sortBy(_._2)
+      .mapPartitions(it => poolAdjacentViolators(it.toArray, isotonic).toIterator)
+
+    poolAdjacentViolators(parallelStepResult.collect(), isotonic)
   }
 }