h, SrSSKJr SSKrSSKrSSKJr SSKJrJr SSKJ r SSK r SSK J r "SS \ \\\ 4\R \\ \ 45rg) z`Implementation of a space that represents the cartesian product of other spaces as a dictionary.) annotationsN) OrderedDict)KeysViewSequence)Any)Spacec^\rSrSrSrSSU4Sjjjr\S5rSSU4SjjjrSSSjjr SSjr SSjr SS jr SS jr S rSS jrSS jrSSjrSSjrSSjrSrU=r$) DictaA dictionary of :class:`Space` instances. Elements of this space are (ordered) dictionaries of elements from the constituent spaces. Example: >>> from gymnasium.spaces import Dict, Box, Discrete >>> observation_space = Dict({"position": Box(-1, 1, shape=(2,)), "color": Discrete(3)}, seed=42) >>> observation_space.sample() {'color': np.int64(0), 'position': array([-0.3991573 , 0.21649833], dtype=float32)} With a nested dict: >>> from gymnasium.spaces import Box, Dict, Discrete, MultiBinary, MultiDiscrete >>> Dict( # doctest: +SKIP ... { ... "ext_controller": MultiDiscrete([5, 2, 2]), ... "inner_state": Dict( ... { ... "charge": Discrete(100), ... "system_checks": MultiBinary(10), ... "job_status": Dict( ... { ... "task": Discrete(5), ... "progress": Box(low=0, high=100, shape=()), ... } ... ), ... } ... ), ... } ... ) It can be convenient to use :class:`Dict` spaces if you want to make complex observations or actions more human-readable. Usually, it will not be possible to use elements of this space directly in learning code. However, you can easily convert :class:`Dict` observations to flat arrays by using a :class:`gymnasium.wrappers.FlattenObservation` wrapper. Similar wrappers can be implemented to deal with :class:`Dict` actions. c >[U[5(a[UR55nO[U[R R 5(a$[[UR555nOF[U[5(a [U5nO%Uc [5nO[S[U535eUR5HupEXA;aXQU'M[SUS35e Xl URR5H'upE[U[5(aMSUSU35e [TU]=SSU5 g![a [UR55nNf=f)aConstructor of :class:`Dict` space. This space can be instantiated in one of two ways: Either you pass a dictionary of spaces to :meth:`__init__` via the ``spaces`` argument, or you pass the spaces as separate keyword arguments (where you will need to avoid the keys ``spaces`` and ``seed``) Args: spaces: A dictionary of spaces. This specifies the structure of the :class:`Dict` space seed: Optionally, you can use this argument to seed the RNGs of the spaces that make up the :class:`Dict` space. **spaces_kwargs: If ``spaces`` is ``None``, you need to pass the constituent spaces as keyword arguments, as described above. NzSUnexpected Dict space input, expecting dict, OrderedDict or Sequence, actual type: zDict space keyword 'z*' already exists in the spaces dictionary.z5Dict space element is not an instance of Space: key='z ', space=) isinstancerdictitems collectionsabcMappingsorted TypeErrorrtype ValueErrorspacesrsuper__init__)selfrseed spaces_kwargskeyspace __class__s O/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/spaces/dict.pyr Dict.__init__6s_" fk * *&,,.)F   7 7 8 8 .fV\\^45  ) )&\F ^VFefjkqfrest  (--/JC #s  *3%/YZ 0.4 ++++-JCu ]Fse9UZT[\ ]. tT*; .flln- .s"E#E87E8cV[SURR555$)zDChecks whether this space can be flattened to a :class:`spaces.Box`.c38# UHoRv M g7fN)is_np_flattenable).0rs r )Dict.is_np_flattenable..psM8Lu**8Ls)allrvaluesrs r r%Dict.is_np_flattenablems"M 8J8J8LMMMc >Uc>URR5VVs0sHup#X#RS5_M snn$[U[5(a[ TU] U5 UR R[R"[R5R[UR5S9n[URR5U5VVVs0sH"uup#nX#R[ U55_M$ snnn$[U[5(aUR5URR5:wa7[!SUR5SURR535eUR5Vs0sH#o"URURX5_M% sn$[#S[%U535es snnfs snnnfs snf)aSeed the PRNG of this space and all subspaces. Depending on the type of seed, the subspaces will be seeded differently * ``None`` - All the subspaces will use a random initial seed * ``Int`` - The integer is used to seed the :class:`Dict` space that is used to generate seed values for each of the subspaces. Warning, this does not guarantee unique seeds for all subspaces, though is very unlikely. * ``Dict`` - A dictionary of seeds for each subspace, requires a seed key for every subspace. This supports seeding of multiple composite subspaces (``Dict["space": Dict[...], ...]`` with ``{"space": {...}, ...}``). Args: seed: An optional int or dictionary of subspace keys to int to seed each PRNG. See above for more details. Returns: A dictionary for the seed values of the subspaces N)sizezThe seed keys: z" are not identical to space keys: z4Expected seed type: dict, int or None, actual type: )rrrr intr np_randomintegersnpiinfoint32maxlenziprkeysrrr)rrrsubspacesubseedssubseedrs r r Dict.seedrs # UHnTUTRU;v M g7fr$r)r&rrxs r r' Dict.contains..s%P=OcqvS!11=Os"%F)r rr9rr))rrGs``r contains Dict.containssK a  1668t{{/?/?/A#APT[[=M=M=OPP Pr-c URU$)z*Get the space that is associated to `key`.rF)rrs r __getitem__Dict.__getitem__s{{3r-c,[UR5$)zReturns the keys of the Dict.)rrr+s r r9 Dict.keyss $$r-cv[U[5(dSUS[U535eX RU'g)z*Set the space that is associated to `key`.zTrying to set zF to Dict space with value that is not a gymnasium space, actual type: N)r rrr)rrvalues r __setitem__Dict.__setitem__sL 5   u C5 fgklqgrfs t u ! Cr-c#8# URShvN gN7f)z+Iterator through the keys of the subspaces.NrFr+s r __iter__ Dict.__iter__s;;s c,[UR5$)zAGives the number of simpler spaces that make up the `Dict` space.)r7rr+s r __len__ Dict.__len__s4;;r-c SSRURR5VVs/sH upU<SU3PM snn5-S-$s snnf)z,Gives a string representation of this space.zDict(z, z: ))joinrr)rrBss r __repr__ Dict.__repr__sM dii4;;;L;L;N O;N41A51#;N OP PSV V OsA cb[U[5=(a URUR:H$)z5Check whether `other` is equivalent to this instance.)r r r)rothers r __eq__ Dict.__eq__s( ud # , u||+ r-c URR5VVVs0sH)up#X#RUVs/sHoDUPM sn5_M+ snnn$s snfs snnnf)zCConvert a batch of samples from this space to a JSONable data type.)rr to_jsonable)rsample_nrrrAs r reDict.to_jsonablesZ #kk//1 1  ""h#GhF3Kh#GH H1  #G sAA AAc URR5VVs0sHup#X#RX5_M nnn[[ [ UR 5555n[U5VVVs/sH+nUR5VVs0sH up'X'U_M snnPM- nnnnU$s snnfs snnfs snnnf)zCConvert a JSONable data type to a batch of samples from this space.)rr from_jsonabler7nextiterr*range) rrfrr dict_of_list n_elementsnrQresults r riDict.from_jsonables#kk//1. 1  $$X]3 31 . d<#6#6#89:; :& &.:-?-?-A B-AzsS(]-A B&   .  C sB2:B>B8$B>8B>rF)NN)rz5None | dict[str, Space] | Sequence[tuple[str, Space]]rz'dict | int | np.random.Generator | Nonerrr$)rzint | dict[str, Any] | Nonereturnzdict[str, int])r?dict[str, Any] | Noner@rsrrzdict[str, Any])rGrrrbool)rstrrr Space[Any])rrr)rrurQrv)rrr0)rrru)rarrrrt)rfzSequence[dict[str, Any]]rrdict[str, list[Any]])rfrwrrzlist[dict[str, Any]])__name__ __module__ __qualname____firstlineno____doc__rpropertyr%rrArIrLr9rRrUrXr^rbreri__static_attributes__ __classcell__)rs@r r r s#NIM8<5+E5+65+ 5+5+nNN%%R'+-1*K#*K+*K  *KX  %!      r-r )r| __future__rcollections.abcrtypingrrrrnumpyr3gymnasium.spaces.spacerrrurr r-r rsNf" #.(t5c3h &..eCj"Atr-