arlbench.core.environments.gymnasium_env

Gymnasium environment adapter.

Classes

GymnasiumEnv(env_name, seed[, env_kwargs])

A gymnasium-based RL environment.

class arlbench.core.environments.gymnasium_env.GymnasiumEnv(env_name, seed, env_kwargs=None)[source]

Bases: Environment

A gymnasium-based RL environment.

property action_space

Action space of the environment.

property observation_space

Observation space of the environment.

reset(rng)[source]

Wraps actual reset for jitting.

Return type:

tuple[None, Union[Array, ndarray, bool_, number]]

step(env_state, action, rng)[source]

Wraps actual step for jitting.

Return type:

tuple[None, tuple[Union[Array, ndarray, bool_, number], Union[Array, ndarray, bool_, number], Union[Array, ndarray, bool_, number], dict]]