h dSrSSKJr SSKrSSKJr SSKJr SSKrSSK J r "SS\ \5r g) zPImplementation of a space that represents the cartesian product of other spaces.) annotationsN)Iterable)Any)Spacec^\rSrSrSrSSU4Sjjjr\S5rSSU4SjjjrSSSjjr SSjr SSjr SS jr SS jr SS jrSS jrSS jrSrU=r$)OneOfaAn exclusive tuple (more precisely: the direct sum) of :class:`Space` instances. Elements of this space are elements of one of the constituent spaces. Example: >>> from gymnasium.spaces import OneOf, Box, Discrete >>> observation_space = OneOf((Discrete(2), Box(-1, 1, shape=(2,))), seed=123) >>> observation_space.sample() # the first element is the space index (Discrete in this case) and the second element is the sample from Discrete (np.int64(0), np.int64(0)) >>> observation_space.sample() # this time the Box space was sampled as index=1 (np.int64(1), array([-0.00711833, -0.7257502 ], dtype=float32)) >>> observation_space[0] Discrete(2) >>> observation_space[1] Box(-1.0, 1.0, (2,), float32) >>> len(observation_space) 2 c@>[U[5(d US35e[U5Ul[ UR5S:dS5eURH-n[U[ 5(aMUS[ U535e [TU]!SSU5 g)apConstructor of :class:`OneOf` space. The generated instance will represent the cartesian product :math:`\text{spaces}[0] \times ... \times \text{spaces}[-1]`. Args: spaces (Iterable[Space]): The spaces that are involved in the cartesian product. seed: Optionally, you can use this argument to seed the RNGs of the ``spaces`` to ensure reproducible sampling. z is not an iterablerz'Empty `OneOf` spaces are not supported.z7 does not inherit from `gymnasium.Space`. Actual Type: N) isinstancertuplespaceslenrtypesuper__init__)selfr seedspace __class__s P/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/spaces/oneof.pyrOneOf.__init__"s&(++Kx7J-KK+Fm 4;;!#N%NN#[[Eu ^OPTUZP[}] ^! tT*c:[SUR55$)zDChecks whether this space can be flattened to a :class:`spaces.Box`.c38# UHoRv M g7fN)is_np_flattenable.0rs r *OneOf.is_np_flattenable..;sD u** s)allr rs rrOneOf.is_np_flattenable8sD DDDrc V>Uc/[TU]S5nU4[SUR55-$[ U[ 5(a[TU]U5nUR R[R"[R5R[UR5S9n[TU]U5 U4[S[URU555-$[ U[[45(a[U5[UR5S-:wa-[S[U5S[UR535e[TU]US54[S [URUSS555-$[!S [#U535e) 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:`Tuple` space that is used to generate seed values for each of the subspaces. Warning, this does not guarantee unique seeds for all the subspaces. * ``Tuple[int, ...]`` - Values used to seed the subspaces, first value seeds the OneOf and subsequent seed the subspaces. This allows the seeding of multiple composite subspaces ``[42, 54, ...]``. Args: seed: An optional int or tuple of ints to seed the OneOf space and subspaces. See above for more details. Returns: A tuple of ints used to seed the OneOf space and subspaces Nc3B# UHoRS5v M g7frrrs rrOneOf.seed..Ns(S{eD)9)9{s)sizec3Z# UH!upUR[U55v M# g7fr)rintrrsubseeds rrr'Vs))&@NE 3w<((&@s)+z`Expects that the subspaces of seeds equals the number of subspaces + 1. Actual length of seeds: z, length of subspaces: rc3H# UHupURU5v M g7frr&r+s rrr'`s"4:T 7##:Ts "z3Expected None, int, or tuple of ints, actual type: )rrr r r r* np_randomintegersnpiinfoint32maxrziplist ValueError TypeErrorr)rr super_seedsubseedsrs rr OneOf.seed=s <d+J=5(St{{(S#SS S c " "d+J~~.."&&S-=/H GL =5)&)$++x&@)$ udm , ,4yC ,q00 vwz{xAwBBYZ]^b^i^iZjYklGLa)+e4:=dkk4PQPR8:T4/ Ed4j\R rcURRS[UR5[R S9nURUnUbUb[ SUSU35eUb[U[5(dS[U535e[U5[UR5:Xd)S[UR5S[U535eURXS9nX54$Ub[U[5(dS [U535e[U5[UR5:Xd)S [UR5S[U535eURX#S 9nX54$UR5nX54$) aGenerates a single random sample inside this space. This method draws independent samples from the subspaces. Args: mask: An optional tuple of optional masks for each of the subspace's samples, expects the same number of masks as spaces probability: An optional tuple of optional probability masks for each of the subspace's samples, expects the same number of probability masks as spaces Returns: Tuple of the subspace's samples r)dtypezIOnly one of `mask` or `probability` can be provided, actual values: mask=z, probability=z/Expected type of `mask` is tuple, actual type: zExpected length of `mask` is z, actual length: )maskz6Expected type of `probability` is tuple, actual type: z$Expected length of `probability` is ) probability) r/r0rr r1int64r7r r rsample)rr>r? subspace_idxsubspacesubspace_samples rrA OneOf.samplehs$~~..q#dkk2B"((.S ;;|,   7[\`[aaop{o|}  e N@d M Nt9 ! ^.s4;;/?.@@QRUVZR[Q\] ^'oo43EoFO,, $U \G[HYGZ[ \{#s ( l5c$++6F5GGXY\]hYiXjk l'oo+:SoTO,,'oo/O,,rcZ[U[5=(a [U5S:H=(a [US[R[ 45=(aV SUSs=:*=(a [UR 5:Os =(a$ UR USRUS5$)z?Return boolean specifying if x is a valid member of this space.rr-)r r rr1r@r*r contains)rxs rrHOneOf.containss q%  1A!  11Q4"((C1 1QqT,,C ,, 1 AaD!**1Q40  rc SSRURVs/sHn[U5PM sn5-S-$s snf)z,Gives a string representation of this space.zOneOf(z, ))joinr str)rss r__repr__OneOf.__repr__s5$))T[[$A[SV[$ABBSHH$As< cUVVs/sH2up#[U5URURU/5S/PM4 snn$s snnf)zCConvert a batch of samples from this space to a JSONable data type.r)r*r to_jsonable)rsample_ni subsamples rrSOneOf.to_jsonablesN #+ "*VT[[^// tuple[Any | None, ...] | Noner?rjrituple[int, Any])rIrkribool)rirN)rTz typing.Sequence[tuple[int, Any]]rilist[list[Any]])rTrmrizlist[tuple[Any, ...]])r_r*riz Space[Any])rir*)rfrrirl)__name__ __module__ __qualname____firstlineno____doc__rpropertyrrrArHrPrSrYr`rcrg__static_attributes__ __classcell__)rs@rrrs,IM+$+F++,EE))Z/359/-+/-3/-  /-b  I 8   " HHrr) rr __future__rtypingcollections.abcrrnumpyr1gymnasium.spaces.spacerrrrr|s.V" $(wHE#JwHr