From 608fa56423b13a384719cdc7c06fd963d9e2a127 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Tue, 1 Aug 2023 17:16:54 -0400 Subject: [PATCH] Fix documentation builds --- docs/src/showcase/bayesian_neural_ode.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/src/showcase/bayesian_neural_ode.md b/docs/src/showcase/bayesian_neural_ode.md index 0a981abd98e..abd4be4cf25 100644 --- a/docs/src/showcase/bayesian_neural_ode.md +++ b/docs/src/showcase/bayesian_neural_ode.md @@ -99,9 +99,9 @@ We sample using 500 warmup samples and 500 posterior samples. ```@example bnode integrator = Leapfrog(find_good_stepsize(h, p)) -prop = AdvancedHMC.NUTS{MultinomialTS, GeneralisedNoUTurn}(integrator) +kernel = HMCKernel(Trajectory{MultinomialTS}(integrator, GeneralisedNoUTurn())) adaptor = StanHMCAdaptor(MassMatrixAdaptor(metric), StepSizeAdaptor(0.45, integrator)) -samples, stats = sample(h, prop, p, 500, adaptor, 500; progress = true) +samples, stats = sample(h, kernel, p, 500, adaptor, 500; progress = true) ``` ## Step 5: Plot diagnostics