Skip to content

Commit 3626a87

Browse files
Fix componentarray handling
1 parent 90691a0 commit 3626a87

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/src/showcase/bayesian_neural_ode.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,10 @@ We use the NUTS sampler with an acceptance ratio of δ= 0.45 in this example. In
119119
We sample using 500 warmup samples and 500 posterior samples.
120120

121121
```@example bnode
122-
integrator = AdvancedHMC.Leapfrog(AdvancedHMC.find_good_stepsize(h, Array(p)))
122+
integrator = AdvancedHMC.Leapfrog(AdvancedHMC.find_good_stepsize(h, p))
123123
kernel = AdvancedHMC.HMCKernel(AdvancedHMC.Trajectory{AdvancedHMC.MultinomialTS}(integrator, AdvancedHMC.GeneralisedNoUTurn()))
124124
adaptor = AdvancedHMC.StanHMCAdaptor(AdvancedHMC.MassMatrixAdaptor(metric), AdvancedHMC.StepSizeAdaptor(0.45, integrator))
125-
samples, stats = AdvancedHMC.sample(h, kernel, Array(p), 500, adaptor, 500; progress = true)
125+
samples, stats = AdvancedHMC.sample(h, kernel, p, 500, adaptor, 500; progress = true)
126126
```
127127

128128
## Step 5: Plot diagnostics

0 commit comments

Comments
 (0)