blacked

bioptim/gui/plot.py

@@ -380,12 +380,14 @@ def legend_without_duplicate_labels(ax):
                 else:
                     nb_subplots = max(
                         [
-                            max(
-                                len(nlp.plot[variable].phase_mappings.to_first.map_idx),
-                                max(nlp.plot[variable].phase_mappings.to_first.map_idx) + 1,
+                            (
+                                max(
+                                    len(nlp.plot[variable].phase_mappings.to_first.map_idx),
+                                    max(nlp.plot[variable].phase_mappings.to_first.map_idx) + 1,
+                                )
+                                if variable in nlp.plot
+                                else 0
                             )
-                            if variable in nlp.plot
-                            else 0
                             for nlp in self.ocp.nlp
                         ]
                     )

tests/shard3/test_initial_condition.py

@@ -336,7 +336,9 @@ def test_initial_guess_error_messages(phase_dynamics):
             objective_functions=objective_functions,
         )
 
-    with pytest.raises(ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_init"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_init"
+    ):
         x_init = InitialGuessList()
         x_init.add("bad_key", [1, 2])
         bio_model = BiorbdModel(biorbd_model_path)
@@ -350,7 +352,9 @@ def test_initial_guess_error_messages(phase_dynamics):
         )
 
     del bio_model  # This is to fix a memory bug
-    with pytest.raises(ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_init"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_init"
+    ):
         u_init = InitialGuessList()
         u_init.add("bad_key", [1, 2])
         bio_model = BiorbdModel(biorbd_model_path)
@@ -361,4 +365,4 @@ def test_initial_guess_error_messages(phase_dynamics):
             phase_time=1,
             u_init=u_init,
             objective_functions=objective_functions,
-        )
+        )

tests/shard4/test_update_bounds_and_init.py

@@ -92,27 +92,36 @@ def test_double_update_bounds_and_init(phase_dynamics):
         ocp.update_bounds(x_init, u_init)
     with pytest.raises(RuntimeError, match="u_bounds should be built from a BoundsList"):
         ocp.update_bounds(None, u_init)
-    with pytest.raises(ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_bounds"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_bounds"
+    ):
         x_bounds = BoundsList()
         x_bounds.add("bad_key", [1, 2])
         ocp.update_bounds(x_bounds, u_bounds)
-    with pytest.raises(ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_bounds"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_bounds"
+    ):
         x_bounds = BoundsList()
         x_bounds["q"] = -np.ones((nq, 1)), np.ones((nq, 1))
         u_bounds = BoundsList()
         u_bounds.add("bad_key", [1, 2])
         ocp.update_bounds(x_bounds, u_bounds)
-    with pytest.raises(ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_init"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a state variable, please check for typos in the declaration of x_init"
+    ):
         x_init = InitialGuessList()
         x_init.add("bad_key", [1, 2])
         ocp.update_initial_guess(x_init, u_init)
-    with pytest.raises(ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_init"):
+    with pytest.raises(
+        ValueError, match="bad_key is not a control variable, please check for typos in the declaration of u_init"
+    ):
         x_init = InitialGuessList()
         x_init["q"] = 0.5 * np.ones((nq, 1))
         u_init = InitialGuessList()
         u_init.add("bad_key", [1, 2])
         ocp.update_initial_guess(x_init, u_init)
 
+
 @pytest.mark.parametrize("phase_dynamics", [PhaseDynamics.SHARED_DURING_THE_PHASE, PhaseDynamics.ONE_PER_NODE])
 def test_update_bounds_and_init_with_param(phase_dynamics):
     def my_parameter_function(bio_model, value, extra_value):