Added standard plots to the Betts example.

csu-hmc · Jul 5, 2024 · e52e320 · e52e320
1 parent 7ca929d
commit e52e320
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/examples-gallery/plot_betts2003.py b/examples-gallery/plot_betts2003.py
@@ -119,6 +119,7 @@ def obj_grad(free):
 
 plt.tight_layout()
 
+# %%
 fig_y2, axes_y2 = plt.subplots(3, 1)
 
 axes_y2[0].plot(time, y2_m, '.k',
@@ -135,4 +136,10 @@ def obj_grad(free):
 
 plt.tight_layout()
 
+# %%
+prob.plot_constraint_violations(solution)
+
+# %%
+prob.plot_trajectories(solution)
+
 plt.show()