diff --git a/src/rpad/partnet_mobility_utils/render/pybullet.py b/src/rpad/partnet_mobility_utils/render/pybullet.py index 4b99c02..1a237c1 100644 --- 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)