fix

kellyguo11 · Aug 1, 2024 · 761c327 · 761c327
1 parent 5ce6625
commit 761c327
Showing 1 changed file with 2 additions and 3 deletions.
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
@@ -23,6 +23,7 @@
 from omni.isaac.core.materials import PhysicsMaterial, PreviewSurface
 from omni.isaac.core.objects import DynamicSphere
 from omni.isaac.core.prims import GeometryPrim, RigidPrim, RigidPrimView
+from omni.isaac.core.utils.extensions import enable_extension
 
 import omni.isaac.lab.terrains as terrain_gen
 from omni.isaac.lab.sim import SimulationContext, build_simulation_context
@@ -242,9 +243,7 @@ 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
-            )
+            enable_extension("omni.kit.primitive.mesh")
             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