From 64febcffb21eb18c4dcc6b3ef3aefbf4cc2fbfc5 Mon Sep 17 00:00:00 2001
From: david-cortes <david.cortes.rivera@gmail.com>
Date: Sun, 3 Dec 2023 19:21:45 +0100
Subject: [PATCH] limit number of threads in tests

---
 R-package/tests/testthat/test_basic.R | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/R-package/tests/testthat/test_basic.R b/R-package/tests/testthat/test_basic.R
index e986186cbb4c..97c1353dc43a 100644
--- a/R-package/tests/testthat/test_basic.R
+++ b/R-package/tests/testthat/test_basic.R
@@ -571,12 +571,13 @@ test_that("Can use multi-output labels with built-in objectives", {
   y <- mtcars$mpg
   x <- as.matrix(mtcars[, -1])
   y_mirrored <- cbind(y, -y)
-  dm <- xgb.DMatrix(x, label = y_mirrored)
+  dm <- xgb.DMatrix(x, label = y_mirrored, nthread = n_threads)
   model <- xgb.train(
     params = list(
       tree_method = "hist",
       multi_strategy = "multi_output_tree",
-      objective = "reg:squarederror"
+      objective = "reg:squarederror",
+      nthread = n_threads
     ),
     data = dm,
     nrounds = 5
@@ -592,7 +593,7 @@ test_that("Can use multi-output labels with custom objectives", {
   y <- mtcars$mpg
   x <- as.matrix(mtcars[, -1])
   y_mirrored <- cbind(y, -y)
-  dm <- xgb.DMatrix(x, label = y_mirrored)
+  dm <- xgb.DMatrix(x, label = y_mirrored, nthread = n_threads)
   model <- xgb.train(
     params = list(
       tree_method = "hist",
@@ -604,7 +605,8 @@ test_that("Can use multi-output labels with custom objectives", {
         hess <- rep(1, nrow(grad) * ncol(grad))
         hess <- matrix(hess, nrow = nrow(grad))
         return(list(grad = grad, hess = hess))
-      }
+      },
+      nthread = n_threads
     ),
     data = dm,
     nrounds = 5