Fix formatting

ARISE-Initiative · Oct 30, 2024 · 125ddc5 · 125ddc5
1 parent aeac5ec
commit 125ddc5
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/tests/test_controllers/test_linear_interpolator.py b/tests/test_controllers/test_linear_interpolator.py
@@ -71,7 +71,7 @@ def step(env, action, current_torques):
         env.sim.forward()
         env._pre_action(action, policy_step)
         last_torques = current_torques
-        current_torques = env.robots[0].composite_controller.part_controllers['right'].torques
+        current_torques = env.robots[0].composite_controller.part_controllers["right"].torques
         summed_abs_delta_torques += np.abs(current_torques - last_torques)
         env.sim.step()
         policy_step = False
@@ -139,7 +139,9 @@ def test_linear_interpolator():
                 initial_state = [env.robots[0]._hand_pos["right"], T.mat2quat(env.robots[0]._hand_orn["right"])]
                 dstate = [
                     env.robots[0]._hand_pos["right"] - initial_state[0],
-                    T.mat2euler(T.quat2mat(T.quat_distance(T.mat2quat(env.robots[0]._hand_orn["right"]), initial_state[1]))),
+                    T.mat2euler(
+                        T.quat2mat(T.quat_distance(T.mat2quat(env.robots[0]._hand_orn["right"]), initial_state[1]))
+                    ),
                 ]
 
                 # Define the uniform trajectory action
@@ -168,7 +170,9 @@ def test_linear_interpolator():
                     timesteps[j] += 1
                     dstate = [
                         env.robots[0]._hand_pos["right"] - initial_state[0],
-                        T.mat2euler(T.quat2mat(T.quat_distance(T.mat2quat(env.robots[0]._hand_orn["right"]), initial_state[1]))),
+                        T.mat2euler(
+                            T.quat2mat(T.quat_distance(T.mat2quat(env.robots[0]._hand_orn["right"]), initial_state[1]))
+                        ),
                     ]
 
                 # When finished, print out the timestep results

diff --git a/tests/test_grippers/test_all_grippers.py b/tests/test_grippers/test_all_grippers.py
@@ -6,6 +6,7 @@
 
 Obviously, if an environment crashes during runtime, that is considered a failure as well.
 """
+
 from robosuite.models.grippers import GRIPPER_MAPPING
 
 

diff --git a/tests/test_grippers/test_panda_gripper.py b/tests/test_grippers/test_panda_gripper.py
@@ -1,6 +1,7 @@
 """
 Tests panda gripper on grabbing task
 """
+
 from robosuite.models.grippers import GripperTester, PandaGripper
 
 

diff --git a/tests/test_grippers/test_rethink_gripper.py b/tests/test_grippers/test_rethink_gripper.py
@@ -1,6 +1,7 @@
 """
 Tests two finger gripper and left two finger gripper on grabbing task
 """
+
 from robosuite.models.grippers import GripperTester, RethinkGripper