hSrSSKJr SSKJr SSKJr SSKrSSK J r SSK J r J r "SS \ \ \R5rg) zOImplementation of a space that consists of binary np.ndarrays of a fixed shape.) annotations)Sequence)AnyN)NDArray) MaskNDArraySpacec^\rSrSrSrS SU4Sjjjr\SSj5r\S5rSSSjjr SSjr SSjr SS jr SS jr SS jrS rU=r$) MultiBinaryaAn n-shape binary space. Elements of this space are binary arrays of a shape that is fixed during construction. Example: >>> from gymnasium.spaces import MultiBinary >>> observation_space = MultiBinary(5, seed=42) >>> observation_space.sample() array([1, 0, 1, 0, 1], dtype=int8) >>> observation_space = MultiBinary([3, 2], seed=42) >>> observation_space.sample() array([[1, 0], [1, 0], [1, 1]], dtype=int8) c>[U[5(aD[U5=UlnU4n[R"U5S:R 5(deO[U[ [R45(aH[SU55=Uln[R"U5S:R 5(deO[S[U535e[TU]1U[RU5 g)a^Constructor of :class:`MultiBinary` space. Args: n: This will fix the shape of elements of the space. It can either be an integer (if the space is flat) or some sort of sequence (tuple, list or np.ndarray) if there are multiple axes. seed: Optionally, you can use this argument to seed the RNG that is used to sample from the space. rc38# UHn[U5v M g7fN)int).0is W/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/spaces/multi_binary.py 'MultiBinary.__init__..0s$7QSVVQszsr&c 2UbUb[SUSU35eUGb5[U[R5(dS[ U535eUR [R :XdSUR 35eURUR:Xd!SURSUR35e[R"US:HUS:H-US :H-5(d S U35e[R"US :HURRSS URUR S 9URUR 55$UGb[U[R5(dS [ U535eUR [R:XdS UR 35eURUR:XdSURSU35e[R"[R"US:US:*55(d SU35eURR!URS9U:*RUR 5$URRSS URUR S 9$)a1Generates a single random sample from this space. A sample is drawn by independent, fair coin tosses (one toss per binary variable of the space). Args: mask: An optional ``np.ndarray`` to mask samples with expected shape of ``space.shape``. For ``mask == 0`` then the samples will be ``0``, for a ``mask == 1`` then the samples will be ``1``. For random samples, using a mask value of ``2``. The expected mask shape is the space shape and mask dtype is ``np.int8``. probability: An optional ``np.ndarray`` to mask samples with expected shape of space.shape where each element represents the probability of the corresponding sample element being a 1. The expected mask shape is the space shape and mask dtype is ``np.float64``. Returns: Sampled values from space zIOnly one of `mask` or `probability` can be provided, actual values: mask=z, probability=z:The expected type of the mask is np.ndarray, actual type: z9The expected dtype of the mask is np.int8, actual dtype: z"The expected shape of the mask is z, actual shape: rz9All values of a mask should be 0, 1 or 2, actual values: )lowhighsizedtypezAThe expected type of the probability is np.ndarray, actual type: zCThe expected dtype of the probability is np.float64, actual dtype: z)The expected shape of the probability is zOAll values of the sample probability should be between 0 and 1, actual values: )r5)rrrrrr6r r*rwhere np_randomintegersrastypefloat64 logical_andrandom)r!mask probabilitys rsampleMultiBinary.sampleCs&   7[\`[aaop{o|}   bjj YKDQUJ<X Y bgg% XJ4::,W X% djj( ]3DJJ>**qqtvvTZZ*X Xr&c N[U[5(a[R"U5n[ [U[R 5=(aQ UR UR :H=(a1 [R"[R"US:HUS:H555$)z?Return boolean specifying if x is a valid member of this space.rr1) rrrarrayboolrr*r logical_or)r!xs rcontainsMultiBinary.containssp a " " A q"** % 6 agg% 6r}}Q!VQ!V45  r&cJ[R"U5R5$)zCConvert a batch of samples from this space to a JSONable data type.)rrCtolist)r!sample_ns r to_jsonableMultiBinary.to_jsonablesxx!((**r&cnUVs/sH#n[R"X R5PM% sn$s snf)zCConvert a JSONable data type to a batch of samples from this space.)rrr6)r!rKr@s r from_jsonableMultiBinary.from_jsonables'=EFX6 6::.XFFFs*2c"SURS3$)z,Gives a string representation of this space.z MultiBinary()rr)s r__repr__MultiBinary.__repr__sdffXQ''r&cb[U[5=(a URUR:H$)z5Check whether `other` is equivalent to this instance.)rr r)r!others r__eq__MultiBinary.__eq__s!%-C$&&EGG2CCr&rSr)rz.NDArray[np.integer[Any]] | Sequence[int] | intr"z int | np.random.Generator | None)returnztuple[int, ...])NN)r>MaskNDArray | Noner?r[rZzNDArray[np.int8])rFrrZrD)rKzSequence[NDArray[np.int8]]rZlist[Sequence[int]])rKr\rZzlist[NDArray[np.int8]])rZstr)rWrrZrD)__name__ __module__ __qualname____firstlineno____doc__rpropertyr*r.r@rGrLrOrTrX__static_attributes__ __classcell__)r$s@rr r s&261 91/114 RV<Y&<Yrls8U"$ 5LD%()LDr&