Need to test size before subsetting

juliasilge · juliasilge · commit 13cbb0ff7dae · 2021-02-09T11:25:11.000-07:00
diff --git a/R/multinom_reg.R b/R/multinom_reg.R
@@ -180,11 +180,10 @@ organize_multnet_class <- function(x, object) {
 
 #' @importFrom vctrs vec_size
 organize_multnet_prob <- function(x, object) {
-  x <- x[,,1]
   if (vec_size(x) > 1) {
-    x <- as_tibble(x)
+    x <- as_tibble(x[,,1])
   } else {
-    x <- tibble::as_tibble_row(x)
+    x <- tibble::as_tibble_row(x[,,1])
   }
   x
 }

Original file line number	Diff line number	Diff line change
`@@ -180,11 +180,10 @@ organize_multnet_class <- function(x, object) {`
`180`	`180`
`181`	`181`	`#' @importFrom vctrs vec_size`
`182`	`182`	`organize_multnet_prob <- function(x, object) {`
`183`		`- x <- x[,,1]`
`184`	`183`	`if (vec_size(x) > 1) {`
`185`		`- x <- as_tibble(x)`
	`184`	`+ x <- as_tibble(x[,,1])`
`186`	`185`	`} else {`
`187`		`- x <- tibble::as_tibble_row(x)`
	`186`	`+ x <- tibble::as_tibble_row(x[,,1])`
`188`	`187`	`}`
`189`	`188`	`x`
`190`	`189`	`}`