Suggestion from code review

tidymodels · Dec 20, 2023 · 1614ab6 · 1614ab6
1 parent 5be1968
commit 1614ab6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/R/proportional_hazards-glmnet.R b/R/proportional_hazards-glmnet.R
@@ -621,12 +621,11 @@ survival_prob_coxnet <- function(object,
   }
 
   if (multi) {
-    res <- tibble::tibble(
+    res_formatted <- tibble::tibble(
       penalty = penalty,
       res_patched = res_patched
     ) %>%
-      tidyr::unnest(cols = res_patched)
-    res_formatted <- res %>%
+      tidyr::unnest(cols = res_patched) %>%
       keep_cols(output, keep_penalty = TRUE) %>%
       tidyr::nest(.pred = c(-.row)) %>%
       dplyr::select(-.row)