audit registrations in extension packages for parallelism compatibility

[simonpcouch]

If a model specification in an extension package uses set_dependency() on its engine package but not on the extension package itself, the model specification will not be available in socket cluster (i.e. plan(multisession)) parallel workers, e.g. tidymodels/bonsai#86. We can likely put together a quick script to check for other engines where this is the case.

[simonpcouch]

With dev bonsai, I see:

library(parsnip)
library(tidyverse)

# read in model flat file
models_known <- parsnip:::model_info_table

# find all engines used in extensions
models_extn <- models_known %>% filter(!is.na(pkg))

# locate them in the parsnip model environment
invisible(
  lapply(parsnip:::extensions(), require, character.only = TRUE)
)
#> Loading required package: baguette
#> Loading required package: censored
#> Loading required package: survival
#> Loading required package: discrim
#> Loading required package: multilevelmod
#> Loading required package: plsmod
#> Loading required package: poissonreg
#> Loading required package: rules
#> Loading required package: bonsai
#> Loading required package: agua

# identify the dependencies that have been registered by each extension package
model_env <- get_model_env()
model_env_pkgs <- 
  as.list(model_env)[grepl("_pkgs", names(model_env))] %>%
  enframe() %>%
  unnest("value") %>%
  mutate(model = gsub("_pkgs", "", name))

# find the difference between ones that register the extension itself as a
# dependency and those that don't
registered <- 
  left_join(
    models_extn,
    model_env_pkgs,
    by = c("model", "mode", "engine")
  ) %>%
  rowwise() %>%
  mutate(
    pkg.x = as.character(pkg.x),
    registered_extn = pkg.x %in% unlist(pkg.y)
  )

all(registered$registered_extn)
#> [1] TRUE

^{Created on 2024-07-15 with reprex v2.1.0}

Squared away, now. :)

[github-actions]

This issue has been automatically locked. If you believe you have found a related problem, please file a new issue (with a reprex: https://reprex.tidyverse.org) and link to this issue.