diff --git a/EpiAware/docs/src/showcase/replications/chatzilena-2019/index.jl b/EpiAware/docs/src/showcase/replications/chatzilena-2019/index.jl index 5d6e3f288..b2297b496 100644 --- a/EpiAware/docs/src/showcase/replications/chatzilena-2019/index.jl +++ b/EpiAware/docs/src/showcase/replications/chatzilena-2019/index.jl @@ -590,7 +590,7 @@ Starting from the initial guess, the MAP point is calculated rapidly in one pass # ╔═╡ 6796ae76-bc2d-4895-ba0a-5e2c23c50dfb map_fit_stoch_mdl = maximum_a_posteriori(stochastic_mdl; - adtype = AutoMooncake(config = nothing), + adtype = ADTypes.AutoMooncake(; config = nothing), initial_params = initial_guess ) @@ -602,7 +602,7 @@ Now we can run NUTS, sampling 1000 posterior draws per chain for 4 chains. # ╔═╡ 156272d7-56c4-4ac4-bf3e-7882f4edc144 chn2 = sample( stochastic_mdl, - NUTS(; adtype = AutoMooncake(config = nothing)), + NUTS(; adtype = ADTypes.AutoMooncake(; config = nothing)), MCMCThreads(), 1000, 4; initial_params = fill(map_fit_stoch_mdl.values.array, 4) ) diff --git a/EpiAware/docs/src/showcase/replications/mishra-2020/index.jl b/EpiAware/docs/src/showcase/replications/mishra-2020/index.jl index a4f1f2ee4..e2211f9a5 100644 --- a/EpiAware/docs/src/showcase/replications/mishra-2020/index.jl +++ b/EpiAware/docs/src/showcase/replications/mishra-2020/index.jl @@ -394,7 +394,7 @@ num_threads = min(10, Threads.nthreads()) inference_method = EpiMethod( pre_sampler_steps = [ManyPathfinder(nruns = 4, maxiters = 100)], sampler = NUTSampler( - adtype = AutoEnzyme(), + adtype = ADTypes.AutoEnzyme(), ndraws = 2000, nchains = num_threads, mcmc_parallel = MCMCThreads()) diff --git a/benchmark/make_epiaware_suite.jl b/benchmark/make_epiaware_suite.jl index 11fe9bcd7..9b3c329dd 100644 --- a/benchmark/make_epiaware_suite.jl +++ b/benchmark/make_epiaware_suite.jl @@ -4,7 +4,7 @@ function make_epiaware_suite(model; check = true, adbackends = [ :forwarddiff, :reversediff, :reversediff_compiled, - ADTypes.AutoMooncake(config = nothing), + ADTypes.AutoMooncake(; config = nothing), ADTypes.AutoEnzyme() ]) suite = prefix_warnings(