Skip to content

lit_sb3.envs.invalid_actions_env

source module lit_sb3.envs.invalid_actions_env

Classes

source class InvalidActionEnvDiscrete(dim: int | None = None, ep_length: int = 100, n_invalid_actions: int = 0)

Bases : IdentityEnv[int]

Identity env with a discrete action space. Supports action masking.

Attributes

  • unwrapped : Env[ObsType, ActType] — Returns the base non-wrapped environment.

  • np_random_seed : int — attr:_np_random_seed that if not set will first initialise with a random int as seed.

  • np_random : np.random.Generator — attr:_np_random that if not set will initialise with a random seed.

Methods

source method InvalidActionEnvDiscrete.action_masks() → list[bool]

source class InvalidActionEnvMultiDiscrete(dims: list[int] | None = None, ep_length: int = 100, n_invalid_actions: int = 0)

Bases : IdentityEnv[np.ndarray]

Identity env with a multidiscrete action space. Supports action masking.

Attributes

  • unwrapped : Env[ObsType, ActType] — Returns the base non-wrapped environment.

  • np_random_seed : int — attr:_np_random_seed that if not set will first initialise with a random int as seed.

  • np_random : np.random.Generator — attr:_np_random that if not set will initialise with a random seed.

Methods

source method InvalidActionEnvMultiDiscrete.action_masks() → list[bool]

source class InvalidActionEnvMultiBinary(dims: int | None = None, ep_length: int = 100, n_invalid_actions: int = 0)

Bases : IdentityEnv[np.ndarray]

Identity env with a multibinary action space. Supports action masking.

Attributes

  • unwrapped : Env[ObsType, ActType] — Returns the base non-wrapped environment.

  • np_random_seed : int — attr:_np_random_seed that if not set will first initialise with a random int as seed.

  • np_random : np.random.Generator — attr:_np_random that if not set will initialise with a random seed.

Methods

source method InvalidActionEnvMultiBinary.action_masks() → list[bool]