[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

get_started/rl/fast_td3/fttd3_module.py

@@ -425,7 +425,8 @@ def projection(
         next_dist = F.softmax(self.forward(obs, actions), dim=1)
         proj_dist = torch.zeros_like(next_dist)
         offset = (
-            torch.linspace(0, (batch_size - 1) * self.num_atoms, batch_size, device=device)
+            torch
+            .linspace(0, (batch_size - 1) * self.num_atoms, batch_size, device=device)
             .unsqueeze(1)
             .expand(batch_size, self.num_atoms)
             .long()

metasim/utils/kinematics.py

@@ -103,7 +103,8 @@ def _solve_ik_jax_with_seed(robot, target_link_index, target_wxyz, target_positi
         ]
 
         sol = (
-            jaxls.LeastSquaresProblem(factors, [joint_var])
+            jaxls
+            .LeastSquaresProblem(factors, [joint_var])
             .analyze()
             .solve(
                 initial_vals=jaxls.VarValues.make([joint_var.with_value(prev_cfg)]),
@@ -130,7 +131,8 @@ def _solve_ik_jax_without_seed(robot, target_link_index, target_wxyz, target_pos
         ]
 
         sol = (
-            jaxls.LeastSquaresProblem(factors, [joint_var])
+            jaxls
+            .LeastSquaresProblem(factors, [joint_var])
             .analyze()
             .solve(
                 verbose=False,