Update docs/src/examples/optimal_control/optimal_control.md

SciML · May 16, 2024 · 05fe73e · 05fe73e
1 parent 27249f2
commit 05fe73e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/src/examples/optimal_control/optimal_control.md b/docs/src/examples/optimal_control/optimal_control.md
@@ -118,7 +118,7 @@ Now let's see what we received:
 
 ```@example neuraloptimalcontrol
 l = loss_adjoint(res3.u)
-cb(res, l)
+cb(res3, l)
 p = plot(solve(remake(prob, p = res3.u), Tsit5(), saveat = 0.01), ylim = (-6, 6), lw = 3)
 plot!(p, ts, [first(first(ann([t], ComponentArray(res3.u, ax), st))) for t in ts],
     label = "u(t)", lw = 3)