fix cmdstan test failures

Author: shmercer

R/adaptive_run.R

@@ -2461,6 +2461,7 @@ adaptive_rank_run_live <- function(state,
   )
   btl_cfg <- .adaptive_btl_resolve_config(state, btl_config)
   btl_cfg$refit_pairs_target <- .adaptive_refit_pairs_target(state, btl_cfg)
+  state$config$btl_config <- btl_cfg
   cfg$refit_pairs_target <- btl_cfg$refit_pairs_target
   cfg$stop_thresholds <- btl_cfg
   state$refit_meta$refit_pairs_target_current <- as.integer(btl_cfg$refit_pairs_target)
@@ -2600,6 +2601,7 @@ adaptive_rank_run_live <- function(state,
 
     refit_out <- maybe_refit_btl(state, config = btl_cfg, fit_fn = fit_fn)
     state <- refit_out$state
+    state$config$btl_config <- refit_out$config
     if (isTRUE(refit_out$refit_performed)) {
       state <- .adaptive_linking_refit_update_state(
         state = state,

tests/testthat/test-5026-adaptive-rank-wrapper.R

@@ -614,6 +614,7 @@ test_that("adaptive_rank wrapper supports link_one_spoke import flow", {
   expect_true(nrow(out$logs$link_stage_log) >= 1L)
   expect_true(all(c("link_transform_policy", "link_transform_state", "reliability_link_global") %in%
     names(out$logs$link_stage_log)))
+  expect_true(is.function(out$state$config$btl_config$cmdstan_fit_fn))
   expect_true("rank_link" %in% names(out$items))
 })
 
@@ -659,6 +660,7 @@ test_that("adaptive_rank wrapper supports link_multi_spoke concurrent flow", {
   expect_true(nrow(out$logs$link_stage_log) >= 2L)
   expect_true(all(c("link_transform_policy", "link_transform_state", "link_epoch_id") %in%
     names(out$logs$link_stage_log)))
+  expect_true(is.function(out$state$config$btl_config$cmdstan_fit_fn))
 })
 
 test_that("adaptive_rank wrapper falls back to rank_raw when linked ranks are unavailable", {