fix pyspark unit test

chouqin · chouqin · commit c41b1b6c7cbf · 2014-10-08T18:33:27.000+08:00
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/model/Node.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/model/Node.scala
@@ -157,7 +157,7 @@ class Node (
     }
     val prefix: String = " " * indentFactor
     if (isLeaf) {
-      prefix + s"Predict: $predict\n"
+      prefix + s"Predict: ${predict.predict}\n"
     } else {
       prefix + s"If ${splitToString(split.get, left=true)}\n" +
         leftNode.get.subtreeToString(indentFactor + 1) +

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ class Node (`
`157`	`157`	`}`
`158`	`158`	`val prefix: String = " " * indentFactor`
`159`	`159`	`if (isLeaf) {`
`160`		`- prefix + s"Predict: $predict\n"`
	`160`	`+ prefix + s"Predict: ${predict.predict}\n"`
`161`	`161`	`} else {`
`162`	`162`	`prefix + s"If ${splitToString(split.get, left=true)}\n" +`
`163`	`163`	`leftNode.get.subtreeToString(indentFactor + 1) +`