From 5ac046a772647beb47c8b5b3d512c82077f1d055 Mon Sep 17 00:00:00 2001 From: Runyu Ding Date: Mon, 30 Dec 2024 10:05:16 +0800 Subject: [PATCH] [style] formatting --- robosuite/robots/mobile_robot.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/robosuite/robots/mobile_robot.py b/robosuite/robots/mobile_robot.py index c220fb31ca..74ab9520e0 100644 --- a/robosuite/robots/mobile_robot.py +++ b/robosuite/robots/mobile_robot.py @@ -268,10 +268,12 @@ def setup_references(self): self._ref_actuator_to_joint_id = np.ones(self.sim.model.nu).astype(np.int32) * (-1) for part_name, actuator_ids in self._ref_actuators_indexes_dict.items(): - self._ref_actuator_to_joint_id[actuator_ids] = np.array([ - self._ref_joints_indexes_dict[part_name].index(self.sim.model.actuator_trnid[i, 0]) - for i in actuator_ids - ]) + self._ref_actuator_to_joint_id[actuator_ids] = np.array( + [ + self._ref_joints_indexes_dict[part_name].index(self.sim.model.actuator_trnid[i, 0]) + for i in actuator_ids + ] + ) def control(self, action, policy_step=False): """