use glmnet formals even with cv_select = TRUE

tlverse · Apr 8, 2021 · 54a695a · 54a695a
1 parent 45ff7d7
commit 54a695a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/R/Lrnr_hal9001.R b/R/Lrnr_hal9001.R
@@ -123,7 +123,10 @@ Lrnr_hal9001 <- R6Class(
 
       # pass in formals of glmnet versus cv.glmnet based on cv_select
       if (args$cv_select) {
-        glmnet_other_valid <- names(formals(glmnet::cv.glmnet))
+        glmnet_other_valid <- union(
+          names(formals(glmnet::cv.glmnet)),
+          names(formals(glmnet::glmnet))
+        )
       } else {
         glmnet_other_valid <- names(formals(glmnet::glmnet))
       }