nit: mappartitions -> map

zhengruifeng · zhengruifeng · commit 9ea6e0073bed · 2019-09-26T10:21:38.000+08:00
diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/impl/GradientBoostedTrees.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/impl/GradientBoostedTrees.scala
@@ -134,12 +134,11 @@ private[spark] object GradientBoostedTrees extends Logging {
       treeWeight: Double,
       tree: DecisionTreeRegressionModel,
       loss: OldLoss): RDD[(Double, Double)] = {
-    data.zip(predictionAndError).mapPartitions { iter =>
-      iter.map { case (Instance(label, _, features), (pred, _)) =>
+    data.zip(predictionAndError).map {
+      case (Instance(label, _, features), (pred, _)) =>
         val newPred = updatePrediction(features, pred, tree, treeWeight)
         val newError = loss.computeError(newPred, label)
         (newPred, newError)
-      }
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -134,12 +134,11 @@ private[spark] object GradientBoostedTrees extends Logging {`
`134`	`134`	`treeWeight: Double,`
`135`	`135`	`tree: DecisionTreeRegressionModel,`
`136`	`136`	`loss: OldLoss): RDD[(Double, Double)] = {`
`137`		`- data.zip(predictionAndError).mapPartitions { iter =>`
`138`		`- iter.map { case (Instance(label, _, features), (pred, _)) =>`
	`137`	`+ data.zip(predictionAndError).map {`
	`138`	`+ case (Instance(label, _, features), (pred, _)) =>`
`139`	`139`	`val newPred = updatePrediction(features, pred, tree, treeWeight)`
`140`	`140`	`val newError = loss.computeError(newPred, label)`
`141`	`141`	`(newPred, newError)`
`142`		`- }`
`143`	`142`	`}`
`144`	`143`	`}`
`145`	`144`