def __init__(
self,
env: SequentialEnvWrapper,
load_params: Dict = None,
lower_bound: float = -10.0,
upper_bound: float = 10.0,
alpha: float = 0.999,
) -> None:
PettingzooWrapper.__init__(self, env)
StandardizeReward.__init__(
self, env, load_params, lower_bound, upper_bound, alpha
)
def from_parallel(par_env):
if isinstance(par_env, to_parallel):
return par_env.aec_env
else:
aec_env = _parallel_env_wrapper(par_env)
ordered_env = OrderEnforcingWrapper(aec_env)
return ordered_env
def parallel_to_aec(par_env):
if isinstance(par_env, aec_to_parallel_wrapper):
return par_env.aec_env
else:
aec_env = parallel_to_aec_wrapper(par_env)
ordered_env = OrderEnforcingWrapper(aec_env)
return ordered_env
def comm_env(**kwargs):
raw_env = base_env.raw_env(**kwargs)
# Set all agents to silent
for agent in raw_env.world.agents:
agent.silent = True
env = AssertOutOfBoundsWrapper(raw_env)
env = OrderEnforcingWrapper(env)
env = CommWrapper(env, comm_dict)
env = pad_observations_v0(env)
env = pad_action_space_v0(env)
env = _PettingZooEnv(env)
return env