From 7688e03eea9e471413abcb640bb30356c53d98e5 Mon Sep 17 00:00:00 2001 From: Jaroslav Sevcik Date: Tue, 27 Feb 2024 00:10:21 -0800 Subject: [PATCH] Remove unnecessary parens --- mjx/mujoco/mjx/_src/smooth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mjx/mujoco/mjx/_src/smooth.py b/mjx/mujoco/mjx/_src/smooth.py index 310aa3ae..496c5dbb 100644 --- a/mjx/mujoco/mjx/_src/smooth.py +++ b/mjx/mujoco/mjx/_src/smooth.py @@ -289,7 +289,7 @@ def solve_m(m: Model, d: Data, x: jax.Array) -> jax.Array: # x <- inv(L') * x for _, vals in sorted(updates_j.items(), reverse=True): j, madr_ij, i = np.array(vals).T - x = x.at[j].add(-(d.qLD[madr_ij] * x[i])) + x = x.at[j].add(-d.qLD[madr_ij] * x[i]) # x <- inv(D) * x x = x * d.qLDiagInv @@ -297,7 +297,7 @@ def solve_m(m: Model, d: Data, x: jax.Array) -> jax.Array: # x <- inv(L) * x for _, vals in sorted(updates_i.items()): i, madr_ij, j = np.array(vals).T - x = x.at[i].add(-(d.qLD[madr_ij] * x[j])) + x = x.at[i].add(-d.qLD[madr_ij] * x[j]) return x