Revert "I think I fixed it :)"

This reverts commit 282a381.
pyomeca · Jul 19, 2023 · af86eb5 · af86eb5
1 parent 282a381
commit af86eb5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bioptim/optimization/optimal_control_program.py b/bioptim/optimization/optimal_control_program.py
@@ -1138,10 +1138,10 @@ def update_bounds(
 
         for nlp in self.nlp:
             for key in nlp.states.keys():
-                if f"{key}_states" in nlp.plot and key in nlp.x_bounds.keys():
+                if f"{key}_states" in nlp.plot and key in nlp.x_bounds:
                     nlp.plot[f"{key}_states"].bounds = nlp.x_bounds[key]
-            for key in nlp.controls.keys():
-                if f"{key}_controls" in nlp.plot and key in nlp.u_bounds.keys():
+            for key in nlp.controls:
+                if f"{key}_controls" in nlp.plot and key in nlp.u_bounds:
                     nlp.plot[f"{key}_controls"].bounds = nlp.u_bounds[key]
 
     def update_initial_guess(