Skip to content

Commit 49882f3

Browse files
committed
Fix bug in energy_fn
1 parent 92ee9a0 commit 49882f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/jsrm/systems/planar_pcs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,7 @@ def energy_fn(params: Dict[str, Array], q: Array, q_d: Array) -> Array:
409409
Returns:
410410
E: total energy of shape ()
411411
"""
412-
T = kinetic_energy_fn(params, q_d)
412+
T = kinetic_energy_fn(params, q, q_d)
413413
U = potential_energy_fn(params, q)
414414
E = T + U
415415

0 commit comments

Comments
 (0)