diff --git a/R/tests/Utils/shared_javapredict_SRF.R b/R/tests/Utils/shared_javapredict_SRF.R index f4bc25b72f..e9fafdd992 100644 --- a/R/tests/Utils/shared_javapredict_SRF.R +++ b/R/tests/Utils/shared_javapredict_SRF.R @@ -57,7 +57,7 @@ if (nrow(prediction1) != nrow(prediction2)) { stop("Number of rows mismatch") } -match <- all(prediction1 == prediction2) +match <- all(prediction1[,2:3] == prediction2[,2:3]) if (! match) { for (i in 1:nrow(prediction1)) { rowmatches <- all(prediction1[i,] == prediction2[i,])