arlbench.core.wrappers

class arlbench.core.wrappers.FlattenObservationWrapper(env)

Bases: Wrapper

Wraps the given environment to flatten its observations.

property observation_space: Box

The flattened observation space of the environment.

Returns:

Flattened obseration space.

Return type:

spaces.Box

reset(rng)

Calls the reset() function of the environment and

flattens the returned observations.

Parameters:

rng (chex.PRNGKey) – Random number generator key.

Returns:

Result of the step function but observations are flattened.

Return type:

tuple[Any, jnp.ndarray]

step(env_state, action, rng)

Calls the step() function of the environment and flattens the returned observations.

Parameters:

• env_state (Any) – The internal environment state.

• action (jnp.ndarray) – The actions to take.

• rng (chex.PRNGKey) – Random number generator key.

Returns:

Result of the step function but observations are flattened.

Return type:

tuple[Any, tuple[jnp.ndarray, jnp.ndarray, jnp.ndarray, dict]]

class arlbench.core.wrappers.Wrapper(env)

Bases: object

Base class for ARLBench wrappers.

Modules

flatten_observation	Observation Flattening Wrapper.
wrapper	Wrapper class.

previous

arlbench.core.running_statistics

next

arlbench.core.wrappers.flatten_observation