slight tweaks

kscalelabs · May 15, 2024 · 9126948 · 9126948
1 parent 14e237b
commit 9126948
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/sim/scripts/create_fixed_torso.py b/sim/scripts/create_fixed_torso.py
@@ -2,12 +2,13 @@
 """This script updates the URDF file to fix the joints of the robot."""
 
 import xml.etree.ElementTree as ET
+
 from sim.stompy.joints import StompyFixed
 
 STOMPY_URDF = "stompy/robot.urdf"
 
 
-def update_urdf():
+def update_urdf() -> None:
     tree = ET.parse(STOMPY_URDF)
     root = tree.getroot()
     stompy = StompyFixed()

diff --git a/sim/scripts/create_mjcf.py b/sim/scripts/create_mjcf.py
@@ -6,7 +6,7 @@
 import mujoco
 
 
-def create_mjcf():
+def create_mjcf() -> None:
     model_xml = mujoco.MjModel.from_xml_path(os.path.join(os.getenv("MODEL_DIR"), "robot.urdf"))
     mujoco.mj_saveLastXML(os.path.join(os.getenv("MODEL_DIR"), "robot.xml"), model_xml)
 

diff --git a/sim/scripts/simulate_mjcf.py b/sim/scripts/simulate_mjcf.py
@@ -4,13 +4,14 @@
 Run with mjpython:
     mjpython mjpython sim/scripts/simulate_mjcf.py
 """
+
 import time
+
 import mujoco
 import mujoco.viewer
 
 from sim.env import stompy_mjcf_path
 
-
 model = mujoco.MjModel.from_xml_path(stompy_mjcf_path())
 data = mujoco.MjData(model)