R/zzz.R

In mlr-org/mlr3extralearners: Extra Learners For mlr3

#' @import checkmate
#' @import paradox
#' @import mlr3misc
#' @importFrom R6 R6Class
#' @importFrom mlr3 mlr_learners LearnerClassif LearnerRegr lrn lrns
#' @importFrom stats predict na.omit setNames formula as.formula
#' @importFrom data.table :=
#' @importFrom methods formalArgs getMethod
#' @importFrom utils getFromNamespace packageVersion capture.output
"_PACKAGE"

utils::globalVariables(c("name", "id", "mlr3_package", "required_packages", ".SD", "S", "K", "calibrator"))

#' @export
mlr3::lrn

#' @export
mlr3::lrns

# nocov start
register_mlr3 = function(libname, pkgname) {
  # get mlr_learners dictionary from the mlr3 namespace
  x = utils::getFromNamespace("mlr_learners", ns = "mlr3")
  # this prevents a very long list of extra learners
  lrns = .extralrns_dict$lrns
  for (i in seq_along(lrns)) {
    x$add(names(lrns)[[i]], lrns[[i]])
  }
}

.onLoad = function(libname, pkgname) { # nolint
  register_namespace_callback(pkgname, "mlr3", register_mlr3)
}

.onUnload = function(libpath) { # nolint
  lrns = .extralrns_dict$lrns
  mlr_learners = mlr3::mlr_learners
  for (i in seq_along(lrns)) {
    mlr_learners$remove(names(lrns)[[i]])
  }
}

# nocov end

leanify_package()