Update pybullet.py

r-pad · Apr 16, 2024 · c29145f · c29145f
1 parent 58dc828
commit c29145f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/rpad/partnet_mobility_utils/render/pybullet.py b/src/rpad/partnet_mobility_utils/render/pybullet.py
@@ -281,6 +281,7 @@ def render(
             None,
         ] = None,
         seed: Optional[int] = None,
+        use_egl: bool = False,
     ) -> PartialPC:
         """Sample a partial pointcloud using the Pybullet GL renderer. Currently only supports
         randomized parameters.
@@ -299,7 +300,7 @@ def render(
         """
         if self._render_env is None:
             self._render_env = PMRenderEnv(
-                pm_obj.obj_dir.name, str(pm_obj.obj_dir.parent)
+                pm_obj.obj_dir.name, str(pm_obj.obj_dir.parent), use_egl=use_egl
             )
 
         rng = np.random.default_rng(seed)