rlightning.env.ale_env¶

ALE environment wrapper.

class rlightning.env.ale_env.ALEEnv(config: Any, worker_index: int | None = 0, preprocess_fn: Callable | None = <function default_env_preprocess_fn>, **kwargs: Any)[source]¶

Bases: BaseEnv

Arcade Learning Environment wrapper.

reset(*args: Any, **kwargs: Any) → EnvRet[source]¶: Reset the environment and return an EnvRet.

step(policy_resp: PolicyResponse) → EnvRet[source]¶: Step the environment with a policy response.