[style] formatting

ARISE-Initiative · Dec 30, 2024 · 5ac046a · 5ac046a
1 parent 45df0d8
commit 5ac046a
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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):
         """