def test_reshape(): base_env = DummyEnv(base_obs, base_obs_space, base_act_spaces) env = reshape_v0(base_env, (64, 3)) obs = env.reset() assert obs.shape == (64, 3) first_obs, _, _, _ = env.step(5) assert np.all(np.equal(first_obs, base_obs.reshape([64, 3])))
def _load_env(self, env_name, pettingzoo_params): from pettingzoo import atari from supersuit import resize_v0, frame_skip_v0, reshape_v0, max_observation_v0 env = importlib.import_module( 'pettingzoo.atari.{}'.format(env_name)).env( obs_type='grayscale_image', **pettingzoo_params) env = max_observation_v0(env, 2) env = frame_skip_v0(env, 4) env = resize_v0(env, 84, 84) env = reshape_v0(env, (1, 84, 84)) return env
def new_continuous_dummy(): base_act_spaces = Box(low=np.float32(0.0), high=np.float32(10.0), shape=[3]) return DummyEnv(base_obs, base_obs_space, base_act_spaces) def new_dummy(): return DummyEnv(base_obs, base_obs_space, base_act_spaces) wrappers = [ supersuit.color_reduction_v0(new_dummy(), "R"), supersuit.resize_v0(dtype_v0(new_dummy(), np.uint8), x_size=5, y_size=10), supersuit.resize_v0(dtype_v0(new_dummy(), np.uint8), x_size=5, y_size=10, linear_interp=True), supersuit.dtype_v0(new_dummy(), np.int32), supersuit.flatten_v0(new_dummy()), supersuit.reshape_v0(new_dummy(), (64, 3)), supersuit.normalize_obs_v0(new_dummy(), env_min=-1, env_max=5.0), supersuit.frame_stack_v1(new_dummy(), 8), supersuit.reward_lambda_v0(new_dummy(), lambda x: x / 10), supersuit.clip_reward_v0(new_dummy()), supersuit.clip_actions_v0(new_continuous_dummy()), supersuit.frame_skip_v0(new_dummy(), 4), supersuit.frame_skip_v0(new_dummy(), (4, 6)), supersuit.sticky_actions_v0(new_dummy(), 0.75), supersuit.delay_observations_v0(new_dummy(), 1), ] @pytest.mark.parametrize("env", wrappers) def test_basic_wrappers(env): env.seed(5)
wrapped_env = pad_action_space_v0(_env) parallel_test.parallel_play_test(wrapped_env) wrappers = [ supersuit.color_reduction_v0(knights_archers_zombies_v4.env(), "R"), supersuit.resize_v0(dtype_v0(knights_archers_zombies_v4.env(), np.uint8), x_size=5, y_size=10), supersuit.resize_v0(dtype_v0(knights_archers_zombies_v4.env(), np.uint8), x_size=5, y_size=10, linear_interp=True), supersuit.dtype_v0(knights_archers_zombies_v4.env(), np.int32), supersuit.flatten_v0(knights_archers_zombies_v4.env()), supersuit.reshape_v0(knights_archers_zombies_v4.env(), (512 * 512, 3)), supersuit.normalize_obs_v0(dtype_v0(knights_archers_zombies_v4.env(), np.float32), env_min=-1, env_max=5.0), supersuit.frame_stack_v1(knights_archers_zombies_v4.env(), 8), supersuit.pad_observations_v0(knights_archers_zombies_v4.env()), supersuit.pad_action_space_v0(knights_archers_zombies_v4.env()), supersuit.black_death_v0(knights_archers_zombies_v4.env()), supersuit.agent_indicator_v0(knights_archers_zombies_v4.env(), True), supersuit.agent_indicator_v0(knights_archers_zombies_v4.env(), False), supersuit.reward_lambda_v0(knights_archers_zombies_v4.env(), lambda x: x / 10), supersuit.clip_reward_v0(knights_archers_zombies_v4.env()), supersuit.clip_actions_v0(prison_v2.env(continuous=True)), supersuit.frame_skip_v0(knights_archers_zombies_v4.env(), 4),
supersuit.resize_v1( dtype_v0(knights_archers_zombies_v10.env(vector_state=False), np.uint8), x_size=5, y_size=10, ), supersuit.resize_v1( dtype_v0(knights_archers_zombies_v10.env(vector_state=False), np.uint8), x_size=5, y_size=10, linear_interp=True, ), supersuit.dtype_v0(knights_archers_zombies_v10.env(), np.int32), supersuit.flatten_v0(knights_archers_zombies_v10.env()), supersuit.reshape_v0(knights_archers_zombies_v10.env(vector_state=False), (512 * 512, 3)), supersuit.normalize_obs_v0(dtype_v0(knights_archers_zombies_v10.env(), np.float32), env_min=-1, env_max=5.0), supersuit.frame_stack_v1(combined_arms_v6.env(), 8), supersuit.pad_observations_v0(simple_world_comm_v2.env()), supersuit.pad_action_space_v0(simple_world_comm_v2.env()), supersuit.black_death_v3(combined_arms_v6.env()), supersuit.agent_indicator_v0(knights_archers_zombies_v10.env(), True), supersuit.agent_indicator_v0(knights_archers_zombies_v10.env(), False), supersuit.reward_lambda_v0(knights_archers_zombies_v10.env(), lambda x: x / 10), supersuit.clip_reward_v0(combined_arms_v6.env()), supersuit.nan_noop_v0(knights_archers_zombies_v10.env(), 0), supersuit.nan_zeros_v0(knights_archers_zombies_v10.env()),