mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-09-07 12:09:58 +00:00
63ea5f2517
* Added Sequence space, updated flatten functions to work with Sequence, Graph. WIP. * Small fixes, added Sequence space to tests * Replace Optional[Any] by Any * Added tests for flattening of non-numpy-flattenable spaces * Return all seeds
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""This module implements various spaces.
|
|
|
|
Spaces describe mathematical sets and are used in Gym to specify valid actions and observations.
|
|
Every Gym environment must have the attributes ``action_space`` and ``observation_space``.
|
|
If, for instance, three possible actions (0,1,2) can be performed in your environment and observations
|
|
are vectors in the two-dimensional unit cube, the environment code may contain the following two lines::
|
|
|
|
self.action_space = spaces.Discrete(3)
|
|
self.observation_space = spaces.Box(0, 1, shape=(2,))
|
|
"""
|
|
from gym.spaces.box import Box
|
|
from gym.spaces.dict import Dict
|
|
from gym.spaces.discrete import Discrete
|
|
from gym.spaces.graph import Graph, GraphInstance
|
|
from gym.spaces.multi_binary import MultiBinary
|
|
from gym.spaces.multi_discrete import MultiDiscrete
|
|
from gym.spaces.sequence import Sequence
|
|
from gym.spaces.space import Space
|
|
from gym.spaces.text import Text
|
|
from gym.spaces.tuple import Tuple
|
|
from gym.spaces.utils import flatdim, flatten, flatten_space, unflatten
|
|
|
|
__all__ = [
|
|
"Space",
|
|
"Box",
|
|
"Discrete",
|
|
"Text",
|
|
"Graph",
|
|
"GraphInstance",
|
|
"MultiDiscrete",
|
|
"MultiBinary",
|
|
"Tuple",
|
|
"Sequence",
|
|
"Dict",
|
|
"flatdim",
|
|
"flatten_space",
|
|
"flatten",
|
|
"unflatten",
|
|
]
|