def env(**kwargs):
env = raw_env(**kwargs)
if env.continuous:
env = wrappers.ClipOutOfBoundsWrapper(env)
else:
env = wrappers.AssertOutOfBoundsWrapper(env)
env = wrappers.OrderEnforcingWrapper(env)
return env
def env(**kwargs):
env = raw_env(**kwargs)
if env.continuous:
default_val = np.zeros((1,))
env = wrappers.ClipOutOfBoundsWrapper(env)
else:
default_val = 1
env = wrappers.AssertOutOfBoundsWrapper(env)
env = wrappers.NanNoOpWrapper(env, default_val, "setting action to {}".format(default_val))
env = wrappers.OrderEnforcingWrapper(env)
return env
def test_unwrapped(name, env_module):
env = env_module.env()
base_env = env.unwrapped
env.reset()
agents = env.agents
if discrete_observation(env, agents):
env = wrappers.AssertOutOfBoundsWrapper(env)
env = wrappers.BaseWrapper(env)
env = wrappers.CaptureStdoutWrapper(env)
if box_observation(env, agents) and box_action(env, agents):
env = wrappers.ClipOutOfBoundsWrapper(env)
env = wrappers.OrderEnforcingWrapper(env)
env = wrappers.TerminateIllegalWrapper(env, 1.0)
if env.metadata["is_parallelizable"]:
env = conversions.aec_to_parallel(env)
env = conversions.parallel_to_aec(env)
env = conversions.turn_based_aec_to_parallel(env)
assert env.unwrapped == base_env, "Unwrapped Test: unequal envs"
def env(**kwargs):
env = raw_env(**kwargs)
env = wrappers.ClipOutOfBoundsWrapper(env)
env = wrappers.NanZerosWrapper(env)
env = wrappers.OrderEnforcingWrapper(env)
return env
def env(**kwargs):
env = raw_env(**kwargs)
env = wrappers.ClipOutOfBoundsWrapper(env)
env = wrappers.NanNoOpWrapper(env, [0, 0, 0], "setting action to 0")
env = wrappers.OrderEnforcingWrapper(env)
return env
