rlightning.env.mujoco_env¶

MuJoCo environment wrapper.

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

Bases: BaseEnv

Gymnasium MuJoCo 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.