diff --git a/src/multiagent_mujoco/mujoco_multi.py b/src/multiagent_mujoco/mujoco_multi.py index b556c02..6d93d00 100755 --- a/src/multiagent_mujoco/mujoco_multi.py +++ b/src/multiagent_mujoco/mujoco_multi.py @@ -183,7 +183,7 @@ def reset(self, **kwargs): return self.get_obs() def render(self, **kwargs): - self.env.render(**kwargs) + return self.env.render(**kwargs) def close(self): raise NotImplementedError