From 6aee1a4c4ff1b3539ad0745656add90cee882906 Mon Sep 17 00:00:00 2001 From: Rujikorn Charakorn Date: Fri, 24 Dec 2021 22:18:14 +0700 Subject: [PATCH] add return to render method this allows mode='rgb_array' and 'depth_array' to return the array as in the original single-agent mujoco gym env --- src/multiagent_mujoco/mujoco_multi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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