Update R/LearnerTorchImage.R

mlr-org · Nov 8, 2024 · 06edf6d · 06edf6d
1 parent 5833a0e
commit 06edf6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/LearnerTorchImage.R b/R/LearnerTorchImage.R
@@ -31,7 +31,7 @@ LearnerTorchImage = R6Class("LearnerTorchImage",
     #' @description
     #' Creates a new instance of this [R6][R6::R6Class] class.
     initialize = function(id, task_type, param_set = ps(), label, optimizer = NULL, loss = NULL,
-      callbacks = list(), packages = c("torchvision"), man, properties = NULL,
+      callbacks = list(), packages = "torchvision", man, properties = NULL,
       predict_types = NULL) {
       properties = properties %??% switch(task_type,
         regr = c(),