wrapper fixes

NAMESPACE

@@ -1,6 +1,9 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(makeWrappedModel,TuneWrapper)
+S3method(predictLearner,HyperoptWrapper)
 S3method(print,ParConfig)
+S3method(trainLearner,HyperoptWrapper)
 export(downloadParConfig)
 export(downloadParConfigs)
 export(generateHyperControl)
@@ -21,6 +24,7 @@ export(getParConfigParVals)
 export(getTaskDictionary)
 export(hyperopt)
 export(makeHyperControl)
+export(makeHyperoptWrapper)
 export(makeParConfig)
 export(setHyperControlMeasures)
 export(setHyperControlMlrControl)

R/HyperoptWrapper.R

@@ -10,7 +10,6 @@
 #' @template arg_parconfig
 #' @template arg_hypercontrol
 #' @template arg_showinfo
-#' @template
 #' @return [\code{\link{Learner}}].
 #' @export
 #' @family tune

tests/testthat/test_HyperoptWrapper.R

@@ -3,9 +3,9 @@ context("HyperoptWrapper")
 test_that("hyperoptWrapper works", {
   mlr::configureMlr(show.info = FALSE, show.learner.output = FALSE)
   lrn = makeLearner("classif.svm")
-  lrn = makeHyperoptWrapper(lrn)
+  lrn2 = makeHyperoptWrapper(lrn)
   task = iris.task
-  res = resample(learner = lrn, task = task, resampling = cv2, extract = getTuneResult)
+  res = resample(learner = lrn2, task = task, resampling = cv2, extract = getTuneResult)
   expect_class(res$extract[[1]], "TuneResult")
   expect_data_frame(getNestedTuneResultsX(res))
   expect_data_frame(getNestedTuneResultsOptPathDf(res))
@@ -23,7 +23,7 @@ test_that("hyperoptWrapper works", {
   par.config = setParConfigParSet(par.config, par.set)
   par.config = setParConfigParVals(par.config, par.vals = list())
   lrn2 = makeHyperoptWrapper(learner = lrn, par.config = par.config, hyper.control = hyper.control)
-  res = resample(learner = lrn2, task = task, resampling = cv2, extract = getTuneResult)
+  res = resample(learner = lrn2, task = sonar.task, resampling = cv2, extract = getTuneResult)
   expect_class(res$extract[[1]], "TuneResult")
   expect_data_frame(getNestedTuneResultsX(res))
   expect_data_frame(getNestedTuneResultsOptPathDf(res))