dots to insepct

integrated-inferences · Oct 21, 2024 · f2d1ba3 · f2d1ba3
1 parent 6c25661
commit f2d1ba3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/R/inspect.R b/R/inspect.R
@@ -172,9 +172,9 @@ grab <- function(model, what = NULL, ...)
   }
 
   if (what %in% .large) {
-    model_sum <- summary(model, include = what)
+    model_sum <- summary(model, include = what, ...)
   } else {
-    model_sum <- summary(model)
+    model_sum <- summary(model, ...)
   }
 
   if (print) print(model_sum, what = what)

diff --git a/tests/testthat/test_inspect.R b/tests/testthat/test_inspect.R
@@ -111,3 +111,13 @@ test_that("inspect outputs correct stanfit", {
   expect_output(inspect(model, what = "stanfit"), "Inference for Stan model: simplexes.")
 })
 
+
+
+test_that("inspect handles dots", {
+  expect_true(make_model("X->Y") |>
+                inspect("prior_hyperparameters", nodes = "Y") |>
+                length() == 4)
+})
+
+
+