bayesian: Simplify and expose the family call

Author: hsbadr

NAMESPACE

@@ -21,6 +21,7 @@ importFrom(rlang,call2)
 importFrom(rlang,enquo)
 importFrom(rlang,eval_tidy)
 importFrom(rlang,expr)
+importFrom(rlang,fn_fmls_names)
 importFrom(stats,sd)
 importFrom(stats,update)
 importFrom(tibble,as_tibble)

R/bayesian.R

@@ -347,6 +347,25 @@ bayesian_fit <- function(formula, data, ...) {
   }
   dots$formula.override <- NULL
 
+  # Simplify and expose the family call
+  if (inherits(dots$family, c("family", "brmsfamily"))) {
+    family_chr <- purrr::map_lgl(dots$family, is.character)
+    family_func <- ifelse(
+      inherits(dots$family, "brmsfamily"),
+      "brmsfamily",
+      dots$family$family
+    )
+    family_fmls <- rlang::fn_fmls_names(
+      get(
+        family_func,
+        mode = "function"
+      )
+    )
+    family_chrs <- intersect(names(family_chr[family_chr]), family_fmls)
+    family_args <- dots$family[family_chrs]
+    dots$family <- rlang::call2(family_func, !!!family_args)
+  }
+
   # Pass extra arguments to Stan
   dots <- append(dots, dots$stan_args)
   dots$stan_args <- NULL

R/bayesian_init.R

@@ -2,7 +2,7 @@
 #' @import parsnip
 #' @importFrom dplyr bind_cols
 #' @importFrom purrr map_lgl
-#' @importFrom rlang !!! abort call2 enquo eval_tidy expr
+#' @importFrom rlang !!! abort call2 enquo eval_tidy expr fn_fmls_names
 #' @importFrom stats sd update
 #' @importFrom tibble as_tibble tibble
 #' @importFrom utils globalVariables getFromNamespace