Merge pull request #1052 from stan-dev/fix-expose-dryRun

`rstan`: Fix `expose_stan_functions` with dryRun
stan-dev · Mar 20, 2023 · 84e4b24 · 84e4b24
2 parents 7aca25b + db404cf
commit 84e4b24
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/rstan/rstan/R/expose_stan_functions.R b/rstan/rstan/R/expose_stan_functions.R
@@ -102,7 +102,6 @@ expose_stan_functions <- function(stanmodel, includes = NULL,
     # workaround for packages with src/install.libs.R
       identical(Sys.getenv("WINDOWS"), "TRUE") &&
       !identical(Sys.getenv("R_PACKAGE_SOURCE"), "") )
-  if (inherits(compiled, "try-error")) stop("Compilation failed!")
   if (!isTRUE(show_compiler_warnings)) {
     sink(type = "output")
     close(zz)
@@ -116,6 +115,7 @@ expose_stan_functions <- function(stanmodel, includes = NULL,
   }
   DOTS <- list(...)
   if (isTRUE(DOTS$dryRun)) return(code)
+  if (inherits(compiled, "try-error")) stop("Compilation failed!")
   ENV <- DOTS$env
   if (is.null(ENV)) ENV <- globalenv()
   for (x in compiled$functions) {

diff --git a/rstan/rstan/tests/testthat/test-expose_dryRun.R b/rstan/rstan/tests/testthat/test-expose_dryRun.R
@@ -0,0 +1,10 @@
+test_that("expose_functions works with dryRun", {
+  funcode <- "
+    functions {
+      real rtn_real(real x) {
+        return x;
+      }
+    }"
+
+  expect_no_error(expose_stan_functions(stanc(model_code = funcode), dryRun = TRUE))
+})