Merge pull request #845 from pariterre/master

scouting
pyomeca · Feb 15, 2024 · aa27237 · aa27237
2 parents 22440f4 + e54be49
commit aa27237
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/bioptim/optimization/stochastic_optimal_control_program.py b/bioptim/optimization/stochastic_optimal_control_program.py
@@ -458,10 +458,7 @@ def get_cov_init(
             # concatenate u_init into a single matrix
             if nlp.control_type == ControlType.CONSTANT:
                 u_guess = np.zeros((0, nlp.ns))
-            elif (
-                nlp.control_type == ControlType.LINEAR_CONTINUOUS
-                or nlp.control_type == ControlType.CONSTANT_WITH_LAST_NODE
-            ):
+            elif nlp.control_type in (ControlType.LINEAR_CONTINUOUS, ControlType.CONSTANT_WITH_LAST_NODE):
                 u_guess = np.zeros((0, nlp.ns + 1))
             elif nlp.control_type == ControlType.NONE:
                 u_guess = np.zeros((0, 0))