Using correct imports

python/mujoco/usd/camera.py

@@ -14,8 +14,7 @@
 # ==============================================================================
 """Camera handling for USD exporter."""
 
-# import mujoco.usd.utils as utils_modules
-import utils as utils_module
+import mujoco.usd.utils as utils_modules
 
 import numpy as np
 

python/mujoco/usd/demo.py

@@ -18,8 +18,7 @@ import argparse
 import pathlib
 
 import mujoco
-# from mujoco.usd import exporter
-import exporter
+from mujoco.usd import exporter
 
 def generate_usd_trajectory(local_args):
   """Generates a USD file given the user arguments."""

python/mujoco/usd/exporter.py

@@ -18,14 +18,10 @@ import os
 from typing import List, Optional
 
 import mujoco
-# import mujoco.usd.camera as camera_module
-# import mujoco.usd.lights as light_module
-# import mujoco.usd.objects as object_module
-# import mujoco.usd.shapes as shapes_module
-import camera as camera_module
-import lights as light_module
-import objects as object_module
-import shapes as shapes_module
+import mujoco.usd.camera as camera_module
+import mujoco.usd.lights as light_module
+import mujoco.usd.objects as object_module
+import mujoco.usd.shapes as shapes_module
 import numpy as np
 from PIL import Image as im
 from PIL import ImageOps

python/mujoco/usd/objects.py

@@ -19,10 +19,8 @@ import collections
 from typing import Optional, Dict, Any
 
 import mujoco
-# import mujoco.usd.shapes as shapes_module
-# import mujoco.usd.utils as utils_modules
-import shapes as shapes_module
-import utils as utils_module
+import mujoco.usd.shapes as shapes_module
+import mujoco.usd.utils as utils_modules
 import numpy as np