Update docs/src/examples/hamiltonian_nn.md

SciML · Jul 30, 2023 · a27dd17 · a27dd17
1 parent 345d7ac
commit a27dd17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/src/examples/hamiltonian_nn.md b/docs/src/examples/hamiltonian_nn.md
@@ -63,7 +63,7 @@ ylabel!("Momentum (p)")
 The HNN predicts the gradients ``(\dot q, \dot p)`` given ``(q, p)``. Hence, we generate the pairs ``(q, p)`` using the equations given at the top. Additionally, to supervise the training, we also generate the gradients. Next, we use Flux DataLoader for automatically batching our dataset.
 
 ```@example hamiltonian
-using Flux, DiffEqFlux, DifferentialEquations, Statistics, Plots, ReverseDiff, Random, IterTools, Lux
+using Flux, DiffEqFlux, DifferentialEquations, Statistics, Plots, ReverseDiff, Random, IterTools, Lux, ComponentArrays
 
 t = range(0.0f0, 1.0f0, length = 1024)
 π_32 = Float32(π)