arlbench.core.environments.gymnax_env

Gymnax environment adapter.

Classes

GymnaxEnv(env_name, n_envs[, env_kwargs])

A gymnax-based RL environment.

class arlbench.core.environments.gymnax_env.GymnaxEnv(env_name, n_envs, env_kwargs=None)[source]

Bases: Environment

A gymnax-based RL environment.

property action_space

Action space of the environment.

property observation_space

Observation space of the environment.

reset(rng)[source]

Resets the environment.

sample_action(rng)[source]

Samples a random action from the action space.

step(env_state, action, rng)[source]

Steps the environment forward.