Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions bioptim/dynamics/dynamics_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,9 +155,7 @@ def torque_driven(
)
)
defects[: dq.shape[0], :] = horzcat(*dq_defects)

tau_var = nlp.states["tau"] if "tau" in nlp.states else nlp.controls["tau"]
defects[dq.shape[0] :, :] = tau_var.mapping.to_first.map(tau) - tau_id
defects[dq.shape[0]:, :] = tau - tau_id

return DynamicsEvaluation(dxdt, defects)

Expand Down Expand Up @@ -639,7 +637,7 @@ def inverse_dynamics(
tau[:, i] = nlp.model.InverseDynamics(q, qdot, qddot, f_ext).to_mx()
else:
tau = nlp.model.InverseDynamics(q, qdot, qddot).to_mx()
return tau_var.mapping.to_first.map(tau)
return tau

@staticmethod
def compute_muscle_dot(nlp: NonLinearProgram, muscle_excitations: Union[MX, SX]):
Expand Down