qgym.spaces.multi_discrete module

This module contains the MultiDiscrete space, i.e., multiple discrete intervals. A sample returns one item from each interval.

Usage:
>>> from qgym.spaces import MultiDiscrete
>>> MultiDiscrete(nvec=[2,3,4])
MultiDiscrete([2 3 4])
class qgym.spaces.multi_discrete.MultiDiscrete(nvec, dtype=<class 'numpy.int32'>, *, rng=None)[source]

Bases: MultiDiscrete

Multi-discrete action/observation space for use in RL environments.

__init__(nvec, dtype=<class 'numpy.int32'>, *, rng=None)[source]

Initialize a multi-discrete space, i.e., multiple discrete intervals of given sizes.

Parameters:

  • nvec (ArrayLike) – Vector containing the upper bound of each discrete interval. The lower bound is always set to 0.

  • dtype (str | type[integer[Any]]) – Type of the values in each interval (default np.int64).

  • rng (Generator | None) – Random number generator to be used in this space, if None a new random number generator will be constructed.