[R] Use R's error stream for printing warnings (#9965)

dmlc · Jan 8, 2024 · 7ff6d44 · 7ff6d44
1 parent b3eb5d0
commit 7ff6d44
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/R-package/src/xgboost_custom.cc b/R-package/src/xgboost_custom.cc
@@ -17,7 +17,11 @@ namespace xgboost {
 ConsoleLogger::~ConsoleLogger() {
   if (cur_verbosity_ == LogVerbosity::kIgnore ||
       cur_verbosity_ <= GlobalVerbosity()) {
-    dmlc::CustomLogMessage::Log(log_stream_.str());
+    if (cur_verbosity_ == LogVerbosity::kWarning) {
+      REprintf("%s\n", log_stream_.str().c_str());
+    } else {
+      dmlc::CustomLogMessage::Log(log_stream_.str());
+    }
   }
 }
 TrackerLogger::~TrackerLogger() {

diff --git a/R-package/tests/testthat/test_basic.R b/R-package/tests/testthat/test_basic.R
@@ -83,7 +83,8 @@ test_that("parameter validation works", {
       bar = "foo"
     )
     output <- capture.output(
-      xgb.train(params = params, data = dtrain, nrounds = nrounds)
+      xgb.train(params = params, data = dtrain, nrounds = nrounds),
+      type = "message"
     )
     print(output)
   }