module moplayground.learning.wrappers

---

class MultiObjectiveEpisodeWrapper

Maintains episode step count and sets done at episode end.

method MultiObjectiveEpisodeWrapper.__init__

__init__(env: brax.envs.base.Env, episode_length: int, action_repeat: int)

---

property MultiObjectiveEpisodeWrapper.action_size

---

property MultiObjectiveEpisodeWrapper.backend

---

property MultiObjectiveEpisodeWrapper.observation_size

---

property MultiObjectiveEpisodeWrapper.unwrapped

---

method MultiObjectiveEpisodeWrapper.reset

reset(rng: jax.Array) → State

---

method MultiObjectiveEpisodeWrapper.step

step(state: brax.envs.base.State, action: jax.Array) → State

---

class MultiObjectiveEvalWrapper

Brax env with eval metrics.

---

property MultiObjectiveEvalWrapper.action_size

---

property MultiObjectiveEvalWrapper.backend

---

property MultiObjectiveEvalWrapper.observation_size

---

property MultiObjectiveEvalWrapper.unwrapped

---

method MultiObjectiveEvalWrapper.reset

reset(rng: jax.Array) → State

---

method MultiObjectiveEvalWrapper.step

step(state: brax.envs.base.State, action: jax.Array) → State