Expands meta-package support for tidymodels

r-lib · Jan 9, 2024 · e2c73bb · e2c73bb
1 parent c0a8f64
commit e2c73bb
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,7 @@
 # downlit (development version)
 
+* Expands meta-package support for tidymodels (#173).
+
 # downlit 0.4.3
 
 * Fix for upcoming R-devel (#169).

diff --git a/R/packages.R b/R/packages.R
@@ -34,10 +34,13 @@ register_attached_packages <- function(packages) {
   options("downlit.attached" = union(packages, getOption("downlit.attached")))
 }
 
+meta_packages <- c("tidyverse", "tidymodels")
 add_depends <- function(packages) {
-  if ("tidyverse" %in% packages && is_installed("tidyverse")) {
-    core <- getNamespace("tidyverse")$core
-    packages <- union(packages, core)
+  for (meta in meta_packages) {
+    if (meta %in% packages && is_installed(meta)) {
+      core <- getNamespace(meta)$core
+      packages <- union(packages, core)
+    }
   }
 
   # add packages attached by depends

diff --git a/tests/testthat/test-packages.R b/tests/testthat/test-packages.R
@@ -55,3 +55,8 @@ test_that("adds tidyverse packages", {
   skip_if_not_installed("tidyverse")
   expect_true("ggplot2" %in% add_depends("tidyverse"))
 })
+
+test_that("adds tidymodels packages", {
+  skip_if_not_installed("tidymodels")
+  expect_true("tune" %in% add_depends("tidymodels"))
+})