Skip to content

Commit e84f2fc

Browse files
committed
Fix the test, we should be looking at the second element instead
1 parent 6ddbf05 commit e84f2fc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mllib/src/test/scala/org/apache/spark/mllib/util/MLUtilsSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ class MLUtilsSuite extends FunSuite with LocalSparkContext {
139139
"Each training+validation set combined should contain all of the data.")
140140
}
141141
// K fold cross validation should only have each element in the validation set exactly once
142-
assert(foldedRdds.map(_._1).reduce((x,y) => x.union(y)).collect().sorted ===
142+
assert(foldedRdds.map(_._2).reduce((x,y) => x.union(y)).collect().sorted ===
143143
data.collect().sorted)
144144
}
145145
}

0 commit comments

Comments
 (0)