Fix argument name in AD integration test

ami-iit · Jan 24, 2025 · df1ecaa · df1ecaa
1 parent b107f31
commit df1ecaa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_automatic_differentiation.py b/tests/test_automatic_differentiation.py
@@ -344,7 +344,7 @@ def step(
             model=model,
             data=data_x0,
             joint_force_references=τ,
-            link_forces=W_f_L,
+            link_forces_inertial=W_f_L,
         )
 
         xf_W_p_B = data_xf.base_position

diff --git a/tests/test_simulations.py b/tests/test_simulations.py
@@ -74,7 +74,7 @@ def test_box_with_external_forces(
         data = js.model.step(
             model=model,
             data=data,
-            link_forces=references.link_forces(model=model, data=data),
+            link_forces_inertial=references._link_forces,
         )
 
     # Check that the box didn't move.
@@ -148,7 +148,7 @@ def test_box_with_zero_gravity(
         data = js.model.step(
             model=model,
             data=data,
-            link_forces=references.link_forces(model=model, data=data),
+            link_forces_inertial=references.link_forces(model=model, data=data),
         )
 
     # Check that the box moved as expected.