make recipe() work with sparse matrices

NEWS.md

@@ -4,6 +4,8 @@
 
 * `recipe()` now works with sparse tibbles. (#1364)
 
+* `recipe()` now works with sparse matrices. (#1364)
+
 # recipes 1.1.0
 
 ## Improvements

R/recipe.R

@@ -12,6 +12,13 @@ recipe <- function(x, ...) {
 #' @rdname recipe
 #' @export
 recipe.default <- function(x, ...) {
+
+  # Doing this here since it should work for all types of Matrix classes
+  if (is_sparse_matrix(x)) {
+    x <- sparsevctrs::coerce_to_sparse_tibble(x)
+    return(recipe(x, ...))
+  }
+
   cli::cli_abort(c(
     x = "{.arg x} should be a data frame, matrix, formula, or tibble.",
     i = "{.arg x} is {.obj_type_friendly {x}}."
@@ -208,6 +215,10 @@ recipe.formula <- function(formula, data, ...) {
     ))
   }
 
+  if (is_sparse_matrix(data)) {
+    data <- sparsevctrs::coerce_to_sparse_tibble(data)
+  }
+
   if (!is_tibble(data)) {
     data <- as_tibble(data)
   }

R/sparsevctrs.R

@@ -1,3 +1,7 @@
 is_sparse_tibble <- function(x) {
   any(vapply(x, sparsevctrs::is_sparse_vector, logical(1)))
 }
+
+is_sparse_matrix <- function(x) {
+  methods::is(x, "sparseMatrix")
+}

tests/testthat/test-sparsevctrs.R

@@ -29,3 +29,33 @@ test_that("recipe() accepts sparse tibbles", {
   )
 })
 
+test_that("recipe() accepts sparse matrices", {
+  skip_if_not_installed("modeldata")
+
+  hotel_data <- sparse_hotel_rates()
+
+  expect_no_condition(
+    rec_spec <- recipe(avg_price_per_room ~ ., data = hotel_data)
+  )
+
+  expect_true(
+    is_sparse_tibble(rec_spec$template)
+  )
+
+  expect_no_condition(
+    rec_spec <- recipe(hotel_data)
+  )
+
+  expect_true(
+    is_sparse_tibble(rec_spec$template)
+  )
+
+  expect_no_condition(
+    rec_spec <- recipe(hotel_data, avg_price_per_room ~ .)
+  )
+
+  expect_true(
+    is_sparse_tibble(rec_spec$template)
+  )
+})
+