let fit_xy() take dgCMatrix input

Author: EmilHvitfeldt

R/convert_data.R

@@ -374,7 +374,11 @@ maybe_matrix <- function(x) {
                         "converted to numeric matrix: {non_num_cols}.")
       rlang::abort(msg)
     }
-    x <- as.matrix(x)
+    if (any(vapply(x, sparsevctrs::is_sparse_vector, logical(1)))) {
+      x <- sparsevctrs::coerce_to_sparse_matrix(x)
+    } else {
+      x <- as.matrix(x)
+    }
   }
   # leave alone if matrix or sparse matrix
   x

R/fit.R

@@ -275,6 +275,10 @@ fit_xy.model_spec <-
       }
     }
 
+    if (allow_sparse(object) && methods::is(x, "sparseMatrix")) {
+      x <- sparsevctrs::coerce_to_sparse_data_frame(x)
+    }
+
     cl <- match.call(expand.dots = TRUE)
     eval_env <- rlang::env()
     eval_env$x <- x
@@ -387,7 +391,7 @@ inher <- function(x, cls, cl) {
 
 check_interface <- function(formula, data, cl, model) {
   inher(formula, "formula", cl)
-  inher(data, c("data.frame", "tbl_spark"), cl)
+  inher(data, c("data.frame", "dgCMatrix", "tbl_spark"), cl)
 
   # Determine the `fit()` interface
   form_interface <- !is.null(formula) & !is.null(data)