guard type.predict subset behind is_missing()

related to #1216, #1218
tidymodels · Sep 26, 2024 · f479034 · f479034
1 parent 0e5fa5c
commit f479034
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/betareg-tidiers.R b/R/betareg-tidiers.R
@@ -79,7 +79,7 @@ augment.betareg <- function(x, data = model.frame(x), newdata = NULL,
                             type.predict, type.residuals, ...) {
   augment_columns(
     x, data, newdata,
-    type.predict = type.predict[1],
+    type.predict = type.predict,
     type.residuals = type.residuals
   )
 }

diff --git a/R/utilities.R b/R/utilities.R
@@ -232,7 +232,7 @@ augment_columns <- function(x, data, newdata = NULL, type, type.predict = type,
     args$newdata <- newdata
   }
   if (!missing(type.predict)) {
-    args$type <- type.predict
+    args$type <- type.predict[1]
   }
   if (!inherits(x, "betareg")) {
     args$se.fit <- se.fit