Fix render_mode for ProcConcatVec envs (#236)

supersuit/vector/concat_vec_env.py

@@ -25,6 +25,7 @@ def __init__(self, vec_env_fns, obs_space=None, act_space=None):
             if not hasattr(vec_envs[i], "num_envs"):
                 vec_envs[i] = SingleVecEnv([lambda: vec_envs[i]])
         self.metadata = self.vec_envs[0].metadata
+        self.render_mode = self.vec_envs[0].render_mode
         self.observation_space = vec_envs[0].observation_space
         self.action_space = vec_envs[0].action_space
         tot_num_envs = sum(env.num_envs for env in vec_envs)

supersuit/vector/markov_vector_wrapper.py

@@ -17,6 +17,7 @@ def __init__(self, par_env, black_death=False):
         """
         self.par_env = par_env
         self.metadata = par_env.metadata
+        self.render_mode = par_env.unwrapped.render_mode
         self.observation_space = par_env.observation_space(par_env.possible_agents[0])
         self.action_space = par_env.action_space(par_env.possible_agents[0])
         assert all(

supersuit/vector/multiproc_vec.py

@@ -98,13 +98,12 @@ def async_loop(
                 elif name == "env_is_wrapped":
                     comp_infos = vec_env.env_is_wrapped(data)
 
-                elif name == "render":
-                    render_result = vec_env.render(data)
-                    if data == "rgb_array":
-                        comp_infos = render_result
-
                 else:
                     raise AssertionError("bad tuple instruction name: " + name)
+            elif instr == "render":
+                render_result = vec_env.render()
+                if vec_env.render_mode == "rgb_array":
+                    comp_infos = render_result
             elif instr == "terminate":
                 return
             else:

supersuit/vector/single_vec_env.py

@@ -6,6 +6,7 @@ class SingleVecEnv:
     def __init__(self, gym_env_fns, *args):
         assert len(gym_env_fns) == 1
         self.gym_env = gym_env_fns[0]()
+        self.render_mode = self.gym_env.render_mode
         self.observation_space = self.gym_env.observation_space
         self.action_space = self.gym_env.action_space
         self.num_envs = 1

test/dummy_aec_env.py

@@ -9,7 +9,7 @@ def __init__(self, observations, observation_spaces, action_spaces):
         super().__init__()
         self._observations = observations
         self._observation_spaces = observation_spaces
-
+        self.render_mode = None
         self.agents = sorted([x for x in observation_spaces.keys()])
         self.possible_agents = self.agents[:]
         self._agent_selector = agent_selector(self.agents)