fixing last typo error

Kev1CO · Apr 22, 2024 · 5c7b763 · 5c7b763
1 parent 34e4f4a
commit 5c7b763
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cocofest/optimization/fes_identification_ocp.py b/cocofest/optimization/fes_identification_ocp.py
@@ -83,8 +83,8 @@ def prepare_ocp(
             use_sx=use_sx,
             ode_solver=ode_solver,
             n_threads=n_threads,
-            pulse_duration=pulse_duration,
-            pulse_intensity=pulse_intensity,
+            fixed_pulse_duration=pulse_duration,
+            fixed_pulse_intensity=pulse_intensity,
         )
 
         OcpFesId._sanity_check_id(

diff --git a/tests/shard1/test_ocp_build.py b/tests/shard1/test_ocp_build.py
@@ -737,7 +737,7 @@ def test_all_ocp_fes_errors():
             n_stim=3,
             n_shooting=10,
             final_time=0.3,
-            objective={"custom_objective": "objective_functions"},
+            objective={"custom": "objective_functions"},
         )
 
     with pytest.raises(TypeError, match="All elements in ObjectiveList must be an Objective type"):
@@ -746,7 +746,7 @@ def test_all_ocp_fes_errors():
             n_stim=3,
             n_shooting=10,
             final_time=0.3,
-            objective={"custom_objective": objective_functions},
+            objective={"custom": objective_functions},
         )
 
     with pytest.raises(TypeError, match="ode_solver must be a OdeSolver type"):