import parallelly

NAMESPACE

@@ -186,6 +186,7 @@ import(paradox)
 importFrom(R6,R6Class)
 importFrom(R6,is.R6)
 importFrom(graphics,plot)
+importFrom(parallelly,availableCores)
 importFrom(stats,contr.treatment)
 importFrom(stats,predict)
 importFrom(stats,rnorm)

R/set_threads.R

@@ -20,24 +20,24 @@
 #' Object to set threads for, e.g. a [Learner].
 #' This object is modified in-place.
 #' @param n (`integer(1)`)\cr
-#' Number of threads to use.
+#' Number of threads to use. Defaults to [parallelly::availableCores()].
 #'
 #' @return Same object as input `x` (changed in-place),
 #' with possibly updated parameter values.
 #' @export
-set_threads = function(x, n = parallelly::availableCores()) {
+set_threads = function(x, n = availableCores()) {
  UseMethod("set_threads")
 }
 
 #' @rdname set_threads
 #' @export
-set_threads.default = function(x, n = parallelly::availableCores()) { # nolint
+set_threads.default = function(x, n = availableCores()) { # nolint
  x
 }
 
 #' @rdname set_threads
 #' @export
-set_threads.Learner = function(x, n = parallelly::availableCores()) { # nolint
+set_threads.Learner = function(x, n = availableCores()) { # nolint
  id = x$param_set$ids(tags = "threads")
  if (length(id)) {
  x$param_set$values = insert_named(x$param_set$values, named_list(id, n))
@@ -47,6 +47,6 @@ set_threads.Learner = function(x, n = parallelly::availableCores()) { # nolint
 
 #' @rdname set_threads
 #' @export
-set_threads.list = function(x, n = parallelly::availableCores()) { # nolint
+set_threads.list = function(x, n = availableCores()) { # nolint
  lapply(x, set_threads, n = n)
 }

R/zzz.R

@@ -8,6 +8,7 @@
 #' @importFrom graphics plot
 #' @importFrom stats predict rnorm runif sd contr.treatment
 #' @importFrom uuid UUIDgenerate
+#' @importFrom parallelly availableCores
 #'
 #' @section Learn mlr3:
 #' * Book on mlr3: \url{https://mlr3book.mlr-org.com}