is_sparse_tibble -> sparsevctrs::has_sparse_elements

tidymodels · Oct 4, 2024 · 183f78f · 183f78f
1 parent 78d7fd2
commit 183f78f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 18 deletions.
diff --git a/R/sparsevctrs.R b/R/sparsevctrs.R
@@ -17,10 +17,6 @@
 #' @name sparse_data
 NULL
 
-is_sparse_tibble <- function(x) {
-  any(vapply(x, sparsevctrs::is_sparse_vector, logical(1)))
-}
-
 is_sparse_matrix <- function(x) {
   methods::is(x, "sparseMatrix")
 }
diff --git a/tests/testthat/test-sparsevctrs.R b/tests/testthat/test-sparsevctrs.R
@@ -9,23 +9,23 @@ test_that("recipe() accepts sparse tibbles", {
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 
   expect_no_condition(
     rec_spec <- recipe(hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 
   expect_no_condition(
     rec_spec <- recipe(hotel_data, avg_price_per_room ~ .)
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 })
 
@@ -42,15 +42,15 @@ test_that("prep() accepts sparse tibbles", {
   )
 
   expect_true(
-    is_sparse_tibble(rec$template)
+    sparsevctrs::has_sparse_elements(rec$template)
   )
 
   expect_no_error(
     rec <- prep(rec_spec, training = hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(rec$template)
+    sparsevctrs::has_sparse_elements(rec$template)
   )
 })
 
@@ -68,15 +68,15 @@ test_that("bake() accepts sparse tibbles", {
   )
 
   expect_true(
-    is_sparse_tibble(res)
+    sparsevctrs::has_sparse_elements(res)
   )
 
   expect_no_error(
     res <- bake(rec_spec, new_data = hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(res)
+    sparsevctrs::has_sparse_elements(res)
   )
 })
 
@@ -91,23 +91,23 @@ test_that("recipe() accepts sparse matrices", {
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 
   expect_no_condition(
     rec_spec <- recipe(hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 
   expect_no_condition(
     rec_spec <- recipe(hotel_data, avg_price_per_room ~ .)
   )
 
   expect_true(
-    is_sparse_tibble(rec_spec$template)
+    sparsevctrs::has_sparse_elements(rec_spec$template)
   )
 })
 
@@ -124,15 +124,15 @@ test_that("prep() accepts sparse matrices", {
   )
 
   expect_true(
-    is_sparse_tibble(rec$template)
+    sparsevctrs::has_sparse_elements(rec$template)
   )
 
   expect_no_error(
     rec <- prep(rec_spec, training = hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(rec$template)
+    sparsevctrs::has_sparse_elements(rec$template)
   )
 })
 
@@ -150,15 +150,15 @@ test_that("bake() accepts sparse matrices", {
   )
 
   expect_true(
-    is_sparse_tibble(res)
+    sparsevctrs::has_sparse_elements(res)
   )
 
   expect_no_error(
     res <- bake(rec_spec, new_data = hotel_data)
   )
 
   expect_true(
-    is_sparse_tibble(res)
+    sparsevctrs::has_sparse_elements(res)
   )
 })