Discrete

lerax.space.Discrete

Bases: AbstractSpace[Int[Array, ''], Bool[Array, ' n']]

A space of finite discrete values.

Attributes:

Name	Type	Description
`n`	`int`	The number of discrete values.

Parameters:

Name	Type	Description	Default
`n`	`int`	The number of discrete values.	required

n: int = n

shape: tuple[int, ...]

flat_size: int

__contains__(x: Any) -> bool

__init__(n: int)

canonical() -> Int[Array, '']

sample(
    *, mask: Bool[ArrayLike, " n"] | None = None, key: Key
) -> Int[Array, ""]

contains(x: Any) -> Bool[Array, '']

__eq__(other: object) -> bool

__repr__() -> str

__hash__() -> int

flatten_sample(
    sample: Int[Array, ""],
) -> Float[Array, " 1"]