Skip to content

Commit f0d890a

Browse files
committed
revert err msg in testsuite
1 parent 038ff58 commit f0d890a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mllib/src/test/scala/org/apache/spark/ml/classification/GBTClassifierSuite.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -317,21 +317,21 @@ class GBTClassifierSuite extends MLTest with DefaultReadWriteTest {
317317
val e: SparkException = intercept[SparkException] {
318318
gbt.fit(df1)
319319
}
320-
assert(e.getMessage.contains("Labels must be integers in range [0, 2)"))
320+
assert(e.getMessage.contains("currently only supports binary classification"))
321321
}
322322
val df2 = getTestData(Seq(0.0, 0.1, 1.0, 0.0))
323323
withClue("Classifier should fail if label is not an integer") {
324324
val e: SparkException = intercept[SparkException] {
325325
gbt.fit(df2)
326326
}
327-
assert(e.getMessage.contains("Labels must be integers in range [0, 2)"))
327+
assert(e.getMessage.contains("currently only supports binary classification"))
328328
}
329329
val df3 = getTestData(Seq(0.0, 2.0, 1.0, 0.0))
330330
withClue("Classifier should fail if label is >= 2") {
331331
val e: SparkException = intercept[SparkException] {
332332
gbt.fit(df3)
333333
}
334-
assert(e.getMessage.contains("Labels must be integers in range [0, 2)"))
334+
assert(e.getMessage.contains("currently only supports binary classification"))
335335
}
336336
}
337337

0 commit comments

Comments
 (0)