Skip to content

Commit ee8037a

Browse files
committed
Fixes incorrect test
1 parent 215f31f commit ee8037a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mllib/src/test/scala/org/apache/spark/ml/tree/impl/AltDTSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ class AltDTSuite extends SparkFunSuite with MLlibTestSparkContext {
256256
val featureIndex = 0
257257
val featureArity = 4
258258
val values = Seq(3.0, 1.0, 0.0, 2.0, 2.0)
259-
val labels = Seq(1, 1, 1, 1, 1, 1, 1).map(_.toDouble)
259+
val labels = Seq(1.0, 1.0, 1.0, 1.0, 1.0)
260260
val impurity = Entropy
261261
val metadata = new AltDTMetadata(numClasses = 2, maxBins = 4, minInfoGain = 0.0, impurity)
262262
val (split, stats) =

0 commit comments

Comments
 (0)