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