Lower tolerance in AD example

hersle · Jan 22, 2025 · a54ae10 · a54ae10
1 parent ba47050
commit a54ae10
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/docs/src/automatic_differentiation.md b/docs/src/automatic_differentiation.md
@@ -15,11 +15,12 @@ To do so, let us write a small wrapper function that calculates the power spectr
 ```@example ad
 using SymBoltz
 M = ΛCDM()
+pars = [M.γ.T₀, M.c.Ω₀, M.b.Ω₀, M.ν.Neff, M.g.h, M.b.rec.Yp, M.h.m, M.I.As, M.I.ns, M.Λ.Ω₀] # TODO: no M.Λ.Ω
+prob0 = CosmologyProblem(M, Dict(pars .=> NaN))
 
 function P(k, θ)
-   pars = Dict([M.γ.T₀, M.c.Ω₀, M.b.Ω₀, M.ν.Neff, M.g.h, M.b.rec.Yp, M.h.m, M.I.As, M.I.ns, M.Λ.Ω₀] .=> θ) # TODO: no Λ
-   prob = CosmologyProblem(M, pars) # TODO: remake
-   sol = solve(prob, k)
+   prob = remake(prob0, Dict(pars .=> θ)) # TODO: remake
+   sol = solve(prob, k; verbose = true, thread = true, ptopts = (alg = SymBoltz.KenCarp4(), reltol = 1e-3))
    return spectrum_matter(sol, k)
 end
 ```