diff --git a/supersuit/vector/single_vec_env.py b/supersuit/vector/single_vec_env.py index bbb1b80..4bb105c 100644 --- a/supersuit/vector/single_vec_env.py +++ b/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 diff --git a/test/dummy_aec_env.py b/test/dummy_aec_env.py index ae2f209..dd07186 100644 --- a/test/dummy_aec_env.py +++ b/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)