hMSrSSKJr SSKJrJrJr SSKJrJ r SSK r SSK J r SSKrSSKJr S SjrS S jr"S S \\ \5rg)zJImplementation of a space that represents closed boxes in euclidean space.) annotations)IterableMappingSequence)Any SupportsFloatN)NDArray)SpacecURS:waM[R"U5[R"U5:Xa[ [R"U55$[ U5$)a3Create a shortened string representation of a numpy array. If arr is a multiple of the all-ones vector, return a string representation of the multiplier. Otherwise, return a string representation of the entire array. Args: arr: The array to represent Returns: A short representation of the array r)sizenpminmaxstr)arrs N/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/spaces/box.pyarray_short_reprrsB xx1}s 3266#; s8Oc[R"[U5[R5=(d. [R"[U5[R5$)zKChecks if a scalar variable is an integer or float (does not include bool).)r issubdtypetypeintegerfloating)vars ris_float_integerr s3 ==cBJJ / X2==cBKK3XXrc^\rSrSrSrS\R S4SU4SjjjrSSjrSSjr \ SSj5r \ S5r SSS jjr SSS jjrSS jrSS jrSS jrSSjrSSjrSU4SjjrSrU=r$)Box%aXA (possibly unbounded) box in :math:`\mathbb{R}^n`. Specifically, a Box represents the Cartesian product of n closed intervals. Each interval has the form of one of :math:`[a, b]`, :math:`(-\infty, b]`, :math:`[a, \infty)`, or :math:`(-\infty, \infty)`. There are two common use cases: * Identical bound for each dimension:: >>> Box(low=-1.0, high=2.0, shape=(3, 4), dtype=np.float32) Box(-1.0, 2.0, (3, 4), float32) * Independent bound for each dimension:: >>> Box(low=np.array([-1.0, -2.0]), high=np.array([2.0, 4.0]), dtype=np.float32) Box([-1. -2.], [2. 4.], (2,), float32) Nc( >Uc [S5e[R"U5Ul[R"UR[R5(dk[R"UR[R 5(d7UR[R :Xd[SURS35eUbu[U[5(d[S[U535e[SU55(d[S[SU5535e[S U55nGO([U[R5(ak[U[R5(aLURUR:wa%[S URS UR35eURnO[U[R5(a URnOr[U[R5(a URnOF[U5(a[U5(aS nO#[S [U5S[U535eX0lUR[R :XaSupgGO[R"UR[R 5(ag[#[R$"UR5R&5n[#[R$"UR5R(5nOf[+[R,"UR5R&5n[+[R,"UR5R(5nUR/X5uUlUlUR5X'5uUlUlUR0RU:wa/[SUR0RSUR35eUR6RU:wa/[SUR6RSUR35e[R:"UR0UR6:5(a%[SUR0SUR635e[=UR05Ul[=UR65Ul [BTU]URURU5 g)aConstructor of :class:`Box`. The argument ``low`` specifies the lower bound of each dimension and ``high`` specifies the upper bounds. I.e., the space that is constructed will be the product of the intervals :math:`[\text{low}[i], \text{high}[i]]`. If ``low`` (or ``high``) is a scalar, the lower bound (or upper bound, respectively) will be assumed to be this value across all dimensions. Args: low (SupportsFloat | np.ndarray): Lower bounds of the intervals. If integer, must be at least ``-2**63``. high (SupportsFloat | np.ndarray]): Upper bounds of the intervals. If integer, must be at most ``2**63 - 2``. shape (Optional[Sequence[int]]): The shape is inferred from the shape of `low` or `high` `np.ndarray`s with `low` and `high` scalars defaulting to a shape of (1,) dtype: The dtype of the elements of the space. If this is an integer type, the :class:`Box` is essentially a discrete space. seed: Optionally, you can use this argument to seed the RNG that is used to sample from the space. Raises: ValueError: If no shape information is provided (shape is None, low is None and high is None) then a value error is raised. Nz6Box dtype must be explicitly provided, cannot be None.zInvalid Box dtype (z.), must be an integer, floating, or bool dtypez2Expected Box shape to be an iterable, actual type=c3~# UH3n[R"[U5[R5v M5 g7fN)r rrr.0dims r Box.__init__..js&O#tCy"**==s;=z;Expected all Box shape elements to be integer, actual type=c38# UHn[U5v M g7fr!)rr"s rr%r&lsWsmrfiX\]`XaXamrc38# UHn[U5v M g7fr!)intr"s rr%r&ps4es#c((er(z4Box low.shape and high.shape don't match, low.shape=z , high.shape=)zBox shape is not specified, therefore inferred from low and high. Expected low and high to be np.ndarray, integer, or float.Actual types low=z, high=)rr+z6Box low.shape doesn't match provided shape, low.shape=z, shape=z8Box high.shape doesn't match provided shape, high.shape=zTBox all low values must be less than or equal to high (some values break this), low=)# ValueErrorr dtyperrrbool_ isinstancer TypeErrorralltuplendarrayshaper_shapefloatfinforrr*iinfo _cast_lowlow bounded_below _cast_highhigh bounded_aboveanyrlow_repr high_reprsuper__init__) selfr:r=r4r-seed dtype_min dtype_max __class__s rrC Box.__init__9s: =UV VXXe_  MM$**bjj 1 1}}TZZ55zzRXX%%djj\1_`   eX..He VOOOOQRWWsmrWsRsQtu 4e44E RZZ ( (Zbjj-I-IyyDJJ& J399+UbcgcmcmbnoIIE RZZ ( (IIE bjj ) )JJE c " "'7'='=E$$(I;gd4j\C (-  :: !#' Iy ]]4::r{{ 3 3bhhtzz2667Ibhhtzz2667IBHHTZZ0445IBHHTZZ0445I(,~~c'E$$$(,(H% 4% 88>>U "HHXX`aeakak`lm  99??e #J499??J[[cdhdndncop  66$((TYY& ' 'fgkgogofppwx|yByBxCD )2 )$))4 TZZ6rc@[U5(Ga[R*[R"URU[ S9:n[R "U5(a[SU35e[R"U5(aFURRS:XaUnO?URRS;a[SU35eOX:a[SUSU35e[R"URXRS9nX4$[U[R5(d[S[U535e[R"UR[R5(dj[R"UR[R 5(d6UR[R":Xd[S UR35e[R$"[R "U55(a[SU35e[R*U:n[R$"[R"U55(a\URRS:XaX![R"U5'OqURRS;a[SU35eOHURUR:wa.[R$"X:5(a[SUSU35e[R"UR[R5(a[R"UR[R5(a[R&"UR5R([R&"UR5R(:a9[*R,R/S URS UR35 UR1UR5U4$) zCasts the input Box low value to ndarray with provided dtype. Args: low: The input box low value dtype_min: The dtype's minimum value Returns: The updated low value and for what values the input is bounded (below) r-z+No low value can be equal to `np.nan`, low=ibuz4Box unsigned int dtype don't support `-np.inf`, low=z1Box low is out of bounds of the dtype range, low=z , min dtype=z=Box low must be a np.ndarray, integer, or float, actual type=zABox low must be a floating, integer, or bool dtype, actual dtype=z*Box low's precision lowered by casting to z, current low.dtype=)rr inffullr4r6isnanr,isneginfr-kindr/r3rrrrr.r?r7 precisiongymloggerwarnastype)rDr:rFr;s rr9 Box._cast_lows C VVGbggdjj#U&KKMxx}} #Nse!TUUS!!::??c)#CZZ__ 2$NseT3 GuLYbXcd''$**cBox high must be a np.ndarray, integer, or float, actual type=z;Box high must be a floating or integer dtype, actual dtype=z+Box high's precision lowered by casting to z, current high.dtype=)rr rQr4r6rPrRr,isposinfr-rTr/r3rrrrr.r?r7rUrVrWrXrY)rDr=rGr>posinfs rr<Box._cast_highs D ! !GGDJJEBRVVKMxx~~ #PQUPV!WXXT""::??c)$DZZ__ 2$NtfU3! I$|\e[fg774::t::>D& &dBJJ// TUYZ^U_T`a djj"++66==RZZ88::) QRVR\R\Q]^'' #PQUPV!WXX 266MM[[&Fvvf~~::??c)#,LZZ__ 2$NtfU3tzz)bffY5E.F.F I$|\e[fg  djj"++66MM$**bkk::HHTZZ(22RXXdjj5I5S5SS A$**Mbcgcmcmbno;;tzz*M9 9rcUR$)z.Has stricter type than gym.Space - never None.)r5rDs rr4 Box.shape7s{{rcg)zDChecks whether this space can be flattened to a :class:`spaces.Box`.Tr`s ris_np_flattenableBox.is_np_flattenable<src[[R"UR55n[[R"UR55nUS:Xa U=(a U$US:XaU$US:XaU$[ SU35e)zChecks whether the box is bounded in some sense. Args: manner (str): One of ``"both"``, ``"below"``, ``"above"``. Returns: If the space is bounded Raises: ValueError: If `manner` is neither ``"both"`` nor ``"below"`` or ``"above"`` bothbelowabovez;manner is not in {'below', 'above', 'both'}, actual value: )boolr r1r;r>r,)rDmannerrhris r is_boundedBox.is_boundedAszRVVD../0RVVD../0 V ?U " w L w LOPVxX rc<Ub"[RRSU35eUb"[RRSU35eURRS:Xa UR OUR R S5S-n[R"UR5nUR)UR)-nUR)UR-nURUR)-nURUR-nURRXURS9XE'URRXwRS9URU-XG'URRXfRS9*X6-XF'URR!URUX8XRS9XH'URRS;a[R""U5n[R$"UR[R&5(aj[R("UR5R*S -n [R("UR5R,S - n UR/XS 9nO[R$"UR[R05(ac[R("UR5R*n [R("UR5R,n UR/XS 9nUR UR5nUR[R2:Xa$UR/URUR S 9nU$) aGenerates a single random sample inside the Box. In creating a sample of the box, each coordinate is sampled (independently) from a distribution that is chosen according to the form of the interval: * :math:`[a, b]` : uniform distribution * :math:`[a, \infty)` : shifted exponential distribution * :math:`(-\infty, b]` : shifted negative exponential distribution * :math:`(-\infty, \infty)` : normal distribution Args: mask: A mask for sampling values from the Box space, currently unsupported. probability: A probability mask for sampling values from the Box space, currently unsupported. Returns: A sampled value from the Box z4Box.sample cannot be provided a mask, actual value: z@Box.sample cannot be provided a probability mask, actual value: fint64r+)r )r:r=r )rLrOrN)rr)rVerrorErrorr-rTr=rYr emptyr4r;r> np_randomnormal exponentialr:uniformfloorr signedintegerr8rrclipunsignedintegerrp) rDmask probabilityr=sample unbounded upp_bounded low_boundedboundedrFrGs rr Box.sampleZs$  ))//FtfM  $))//RS^R_` !JJOOs2tyy 8H8H8QTU8U$**%'''4+=+=*== )))D,>,>> ((D,>,>+>> $$t'9'99!NN11y7K7Q7Q1R NN & &K,D,J,J & Khh{# $  ^^ ' '[-E-K-K ' L L   ..00! GF ]]4::r'9'9 : :,00I,00I[[Y[>Ftzz* :: ![[TXX499[=F rc4[U[R5(d>[RR S5 [R "XRS9n[[R"URUR5=(ak URUR:H=(aK [R"XR:5=(a" [R"XR:*55$![[4a gf=f)z?Return boolean specifying if x is a valid member of this space.zCasting input x to numpy array.rKF)r/r r3rVrWrXasarrayr-r,r0rjcan_castr4r1r:r=)rDxs rcontains Box.containss!RZZ(( JJOO= > JJq 3 KK , '4::% 'qHH}% 'qII~&    *  sDDDcLUVs/sHo"R5PM sn$s snf)zCConvert a batch of samples from this space to a JSONable data type.)tolistrDsample_nrs r to_jsonableBox.to_jsonables.67hF h777s!cjUVs/sH!n[R"X RS9PM# sn$s snf)zCConvert a JSONable data type to a batch of samples from this space.rK)r rr-rs r from_jsonableBox.from_jsonables'CKL8 648LLLs(0c pSURSURSURSURS3 $)zA string representation of this space. The representation will include bounds, shape and dtype. If a bound is uniform, only the corresponding scalar will be given to avoid redundant and ugly strings. Returns: A representation of the space zBox(z, ))r@rAr4r-r`s r__repr__ Box.__repr__s7dmm_Bt~~&6b BtzzlRSTTrcj[U[5=(a URUR:H=(a} URUR:H=(a] [R "UR UR 5=(a+ [R "URUR5$)zVCheck whether `other` is equivalent to this instance. Doesn't check dtype equivalence.)r/rr4r-r allcloser:r=)rDothers r__eq__ Box.__eq__sw uc " 3u{{* 3u{{* 3 DHHeii0 3 DIIuzz2  rc>[TU]U5 [US5(d[UR5Ul[US5(d[UR 5Ulgg)zCSets the state of the box for unpickling a box with legacy support.r@rAN)rB __setstate__hasattrrr:r@r=rA)rDstaterHs rrBox.__setstate__sQ U#tZ((,TXX6DMt[))-dii8DN*r)r5r>r;r-r=rAr:r@) r:SupportsFloat | NDArray[Any]r=rr4zSequence[int] | Noner-z.type[np.floating[Any]] | type[np.integer[Any]]rEz int | np.random.Generator | None)returnztuple[np.ndarray, np.ndarray])rztuple[int, ...])rg)rkrrrj)NN)r}Noner~rr NDArray[Any])rrrrj)rzSequence[NDArray[Any]]rz list[list])rzSequence[float | int]rzlist[NDArray[Any]])rr)rrrrj)rz-Iterable[tuple[str, Any]] | Mapping[str, Any])__name__ __module__ __qualname____firstlineno____doc__r float32rCr9r<propertyr4rdrlrrrrrrr__static_attributes__ __classcell__)rHs@rrr%s.'+@B 15 q7 )q7+q7$ q7 > q7 / q7q7fC9JD:L2IV 8M U  9 9rr)rrrr)rrrrj)r __future__rcollections.abcrrrtypingrrnumpyr numpy.typingr gymnasiumrVgymnasium.spaces.spacer rrrrcrrrsBP"77% ("Y v9% v9r