arlbench.core.environments.envpool_env

Envpool environment adapter.

Functions

numpy_to_jax(x)

Converts numpy arrays to jax numpy arrays.

Classes

EnvpoolEnv(env_name, n_envs, seed[, env_kwargs])

An envpool-based RL environment.

class arlbench.core.environments.envpool_env.EnvpoolEnv(env_name, n_envs, seed, env_kwargs=None)[source]

Bases: Environment

An envpool-based RL environment.

property action_space

Action space of the environment.

property observation_space

Observation space of the environment.

reset(_)[source]

Resets the environment.

step(env_state, action, _)[source]

Steps the environment forward by one step.

arlbench.core.environments.envpool_env.numpy_to_jax(x)[source]

Converts numpy arrays to jax numpy arrays.

Return type:

Array