fix assertation message

Farama-Foundation · Aug 15, 2022 · aa6e667 · aa6e667
1 parent 7678238
commit aa6e667
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