R/addons-shapr.R

In futurize: Parallelize Common Functions via One Magic Function

# shapr::explain(...) =>
#
# local({
#   ## This will be automatically consumed and removed by 'future.apply'
#   options(future.disposable = structure(<future options>, dispose = FALSE))
#   on.exit(options(future.disposable = NULL))
#   shapr::explain(...)
# })
#
append_transpilers_for_shapr <- function() {
  transpilers <- make_package_transpilers("shapr", FUN = function(fcn, name) {
    if (name %in% c("explain", "explain_forecast")) {
      list(
        label = sprintf("shapr::%s() ~> shapr::%s()", name, name),
        transpiler = make_futurize_for_future.apply(
          defaults = list(
            future.seed = TRUE
          )
        )
      )
    }
  })

  append_transpilers("futurize::add-on", transpilers)

  ## Return required packages
  c("shapr", "future.apply")
}