fix test

kellyguo11 · Jul 31, 2024 · 7645a27 · 7645a27
1 parent a282d82
commit 7645a27
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/source/extensions/omni.isaac.lab/test/terrains/check_terrain_importer.py b/source/extensions/omni.isaac.lab/test/terrains/check_terrain_importer.py
@@ -65,6 +65,7 @@
 import numpy as np
 
 import omni.isaac.core.utils.prims as prim_utils
+import omni.kit
 import omni.kit.commands
 from omni.isaac.cloner import GridCloner
 from omni.isaac.core.materials import PhysicsMaterial, PreviewSurface
@@ -79,6 +80,8 @@
 from omni.isaac.lab.terrains.terrain_importer import TerrainImporter
 from omni.isaac.lab.utils.assets import ISAAC_NUCLEUS_DIR
 
+omni.kit.app.get_app().get_extension_manager().set_extension_enabled_immediate("omni.kit.primitive.mesh", True)
+
 
 def main():
     """Generates a terrain from isaaclab."""

diff --git a/source/extensions/omni.isaac.lab/test/terrains/test_terrain_importer.py b/source/extensions/omni.isaac.lab/test/terrains/test_terrain_importer.py
@@ -17,6 +17,7 @@
 import unittest
 
 import omni.isaac.core.utils.prims as prim_utils
+import omni.kit
 import omni.kit.commands
 from omni.isaac.cloner import GridCloner
 from omni.isaac.core.materials import PhysicsMaterial, PreviewSurface
@@ -241,6 +242,9 @@ def _populate_scene(self, sim: SimulationContext, num_balls: int = 2048, geom_sp
             )
         else:
             # -- Ball geometry
+            omni.kit.app.get_app().get_extension_manager().set_extension_enabled_immediate(
+                "omni.kit.primitive.mesh", True
+            )
             cube_prim_path = omni.kit.commands.execute("CreateMeshPrimCommand", prim_type="Sphere")[1]
             prim_utils.move_prim(cube_prim_path, "/World/envs/env_0/ball")
             # -- Ball physics