Merge pull request #964 from stan-dev/pathfinder-init-funs

andrjohns · web-flow · commit c3d01688dce9 · 2024-05-07T22:07:59.000+03:00
Pathfinder method errors with init function due to NULL default num_paths
diff --git a/R/model.R b/R/model.R
@@ -1939,7 +1939,7 @@ pathfinder <- function(data = NULL,
                        history_size = NULL,
                        single_path_draws = NULL,
                        draws = NULL,
-                       num_paths = NULL,
+                       num_paths = 4,
                        max_lbfgs_iters = NULL,
                        num_elbo_draws = NULL,
                        save_single_paths = NULL,
diff --git a/man/model-method-pathfinder.Rd b/man/model-method-pathfinder.Rd
diff --git a/tests/testthat/test-model-pathfinder.R b/tests/testthat/test-model-pathfinder.R
@@ -124,6 +124,11 @@ test_that("pathfinder() method works with init file", {
   expect_pathfinder_output(mod$pathfinder(data = data_file_r, init = init_file))
 })
 
+test_that("pathfinder() method works with init function and default paths", {
+  init_function <- function() { list(theta = 0.5) }
+  expect_pathfinder_output(mod$pathfinder(data = data_file_r, init = init_function))
+})
+
 test_that("pathfinder() method runs when all arguments specified", {
   expect_pathfinder_output(fit <- do.call(mod$pathfinder, ok_arg_values))
   expect_is(fit, "CmdStanPathfinder")
