From c29145fa394eab3f3b32fe69c090e71cc352606b Mon Sep 17 00:00:00 2001 From: Ben Eisner Date: Tue, 16 Apr 2024 14:05:31 -0400 Subject: [PATCH] Update pybullet.py --- src/rpad/partnet_mobility_utils/render/pybullet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)