Skip to content

Commit

Permalink
Fix the test, we should be looking at the second element instead
Browse files Browse the repository at this point in the history
  • Loading branch information
holdenk committed Apr 10, 2014
1 parent 6ddbf05 commit e84f2fc
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ class MLUtilsSuite extends FunSuite with LocalSparkContext {
"Each training+validation set combined should contain all of the data.")
}
// K fold cross validation should only have each element in the validation set exactly once
assert(foldedRdds.map(_._1).reduce((x,y) => x.union(y)).collect().sorted ===
assert(foldedRdds.map(_._2).reduce((x,y) => x.union(y)).collect().sorted ===
data.collect().sorted)
}
}
Expand Down

0 comments on commit e84f2fc

Please sign in to comment.