`evorl.envs.gymnax`¶

Module Contents¶

Classes¶

GymnaxAdapter

Adapter for Gymnax environments.

Functions¶

`create_gymnax_env`
`create_wrapped_gymnax_env`
`gymnax_space_to_evorl_space`

API¶

class evorl.envs.gymnax.GymnaxAdapter(env: gymnax.environments.environment.Environment, env_params: chex.ArrayTree | None = None)¶

Bases: evorl.envs.env.EnvAdapter

Adapter for Gymnax environments.

property action_space: evorl.envs.space.Space¶

property obs_space: evorl.envs.space.Space¶

reset(key: chex.PRNGKey) → evorl.envs.env.EnvState¶

step(state: evorl.envs.env.EnvState, action: evorl.types.Action) → evorl.envs.env.EnvState¶

evorl.envs.gymnax.create_gymnax_env(env_name: str, flatten_obs: bool = True, **kwargs) → evorl.envs.gymnax.GymnaxAdapter¶

evorl.envs.gymnax.create_wrapped_gymnax_env(env_name: str, flatten_obs: bool = True, episode_length: int = 1000, parallel: int = 1, autoreset_mode: evorl.envs.wrappers.training_wrapper.AutoresetMode = AutoresetMode.NORMAL, discount: float | None = 1.0, record_ori_obs: bool = False, **kwargs) → evorl.envs.env.Env¶

evorl.envs.gymnax.gymnax_space_to_evorl_space(space: gymnax.environments.spaces.Space)¶