fix data index

SciML · May 20, 2024 · f871f8d · f871f8d
1 parent 556ee66
commit f871f8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/src/showcase/bayesian_neural_ode.md b/docs/src/showcase/bayesian_neural_ode.md
@@ -71,14 +71,14 @@ p = ComponentArray{Float64}(p)
 const _p = p
 ```
 
-Note that the `f64` is required to put the Flux neural network into Float64 precision.
+Note that the `f64` is required to put the Lux neural network into Float64 precision.
 
 ## Step 3: Define the loss function for the Neural ODE.
 
 ```@example bnode
 function predict_neuralode(p)
     p = p isa ComponentArray ? p : convert(typeof(_p),p)
-    Array(prob_neuralode(u0, p)[1])
+    Array(prob_neuralode(u0, p))
 end
 function loss_neuralode(p)
     pred = predict_neuralode(p)