diff --git a/bigym/bigym_renderer.py b/bigym/bigym_renderer.py index 3cd45ae..f6c05f1 100644 --- a/bigym/bigym_renderer.py +++ b/bigym/bigym_renderer.py @@ -21,17 +21,32 @@ class BiGymRenderer(MujocoRenderer): def __init__(self, mojo: Mojo): """Init.""" - super().__init__(mojo.model, mojo.data) + super().__init__( + mojo.model, + mojo.data, + width=mojo.model.vis.global_.offwidth, + height=mojo.model.vis.global_.offheight, + ) def _get_viewer(self, render_mode: str) -> BaseRender: """See base.""" self.viewer = self._viewers.get(render_mode) if self.viewer is None: if render_mode == "human": - self.viewer = BiGymWindowViewer(self.model, self.data) + self.viewer = BiGymWindowViewer( + self.model, + self.data, + width=self.width, + height=self.height, + ) elif render_mode in {"rgb_array", "depth_array"}: - self.viewer = OffScreenViewer(self.model, self.data) + self.viewer = OffScreenViewer( + self.model, + self.data, + width=self.width, + height=self.height, + ) else: raise AttributeError( f"Unexpected mode: {render_mode}, " diff --git a/setup.py b/setup.py index a4439d1..02addde 100644 --- a/setup.py +++ b/setup.py @@ -21,8 +21,7 @@ def get_version(rel_path): core_requirements = [ - # includes bugfix in mujoco_rendering - "gymnasium @ git+https://git@github.com/stepjam/Gymnasium.git@0.29.2", + "gymnasium>=1.0.0", # pyquaternion doesn't support 2.x yet "numpy==1.26.*", "safetensors==0.6.2",