inherit predict_sets from base learner in AutoFSelector

mllg · mllg · commit 31fa444d05d4 · 2020-10-28T10:42:13.000+01:00
diff --git a/R/AutoFSelector.R b/R/AutoFSelector.R
@@ -116,6 +116,7 @@ AutoFSelector = R6Class("AutoFSelector",
       )
 
       self$predict_type = learner$predict_type
+      self$predict_sets = learner$predict_sets
     }
   ),
 

Original file line number	Diff line number	Diff line change
`@@ -116,6 +116,7 @@ AutoFSelector = R6Class("AutoFSelector",`
`116`	`116`	`)`
`117`	`117`
`118`	`118`	`self$predict_type = learner$predict_type`
	`119`	`+ self$predict_sets = learner$predict_sets`
`119`	`120`	`}`
`120`	`121`	`),`
`121`	`122`