Merge pull request #174 from jjshoots/assertation-fix

Fix out of date assertation message
Farama-Foundation · Aug 15, 2022 · d53a76b · d53a76b
2 parents 7678238 + aa6e667
commit d53a76b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/supersuit/vector/markov_vector_wrapper.py b/supersuit/vector/markov_vector_wrapper.py
@@ -22,11 +22,11 @@ def __init__(self, par_env, black_death=False):
         assert all(
             self.observation_space == par_env.observation_space(agent)
             for agent in par_env.possible_agents
-        ), "observation spaces not consistent. Perhaps you should wrap with `supersuit.aec_wrappers.pad_observations`?"
+        ), "observation spaces not consistent. Perhaps you should wrap with `supersuit.multiagent_wrappers.pad_observations_v0`?"
         assert all(
             self.action_space == par_env.action_space(agent)
             for agent in par_env.possible_agents
-        ), "action spaces not consistent. Perhaps you should wrap with `supersuit.aec_wrappers.pad_actions`?"
+        ), "action spaces not consistent. Perhaps you should wrap with `supersuit.multiagent_wrappers.pad_action_space_v0`?"
         self.num_envs = len(par_env.possible_agents)
         self.black_death = black_death