h%x%SrSSKJr SSKJr SSKrSSKJr SSK J r \ "S5r S\ S '"S S \ \5rg) z:Implementation of a space that represents textual strings.) annotations)AnyN)NDArray)Space>abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789frozenset[str] alphanumericc^\rSrSrSrS\SS.SU4SjjjrSSSjjrSSjrSS jr SS jr SS jr \ SS j5r \ SS j5rSSjr\ SSj5r\ SSj5rSrU=r$)TextaA space representing a string comprised of characters from a given charset. Example: >>> from gymnasium.spaces import Text >>> # {"", "B5", "hello", ...} >>> Text(5) Text(1, 5, charset=0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz) >>> # {"0", "42", "0123456789", ...} >>> import string >>> Text(min_length = 1, ... max_length = 10, ... charset = string.digits) Text(1, 10, charset=0123456789) N) min_lengthcharsetseedc>[R"[U5[R5(dS[U535e[R"[U5[R5(dS[U535eSU::d SU35eX!::d SUSU35e[ U5Ul[ U5Ul[U5Ul[U5Ul [[U55VVs0sHupVU[R"U5_M snnUl SR[[U555Ul["TU]I[&US9 g s snnf) aConstructor of :class:`Text` space. Both bounds for text length are inclusive. Args: min_length (int): Minimum text length (in characters). Defaults to 1 to prevent empty strings. max_length (int): Maximum text length (in characters). charset (Union[set], str): Character set, defaults to the lower and upper english alphabet plus latin digits. seed: The seed for sampling from the space. z6Expects the min_length to be an integer, actual type: z6Expects the max_length to be an integer, actual type: rz8Minimum text length must be non-negative, actual value: zIThe min_length must be less than or equal to the max_length, min_length: z, max_length: )dtyperN)np issubdtypetypeintegerintr max_length frozenset _char_settuple _char_list enumerateint32 _char_indexjoinsorted _char_strsuper__init__str)selfrrrrival __class__s O/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/spaces/text.pyr% Text.__init__"sm$}}  bjj   W CDDTCU V W }}  bjj   W CDDTCU V W  O S Ej\ R S   $ ~ VWaVbbpq{p| } ~ $ #:":)27);+0>+4U7^+D1 +DC! +D1 !ggfU7^&<= s. 1 s#E.cUbUb[SUSU35eUbURU[RS5up4Ubs[R"[R "US:HUS:H55(d SU35e[R "U5S:aU[R "U5- nOUbURU[RS5up4Ub[R"[R"US:US:*55(d S U35e[R"[R "U5S5(dS [R "U535eOS=p4Uc3URRURURS-5nUc?[R"[UR 55[UR 5- n[R"US:H5(a*URS:Xag [S URS 35eURR#UR$X4S9nS R'U5$)a^Generates a single random sample from this space with by default a random length between ``min_length`` and ``max_length`` and sampled from the ``charset``. Args: mask: An optional tuples of length and mask for the text. The length is expected to be between the ``min_length`` and ``max_length``. Otherwise, a random integer between ``min_length`` and ``max_length`` is selected. For the mask, we expect a numpy array of length of the charset passed with ``dtype == np.int8``. If the charlist mask is all zero then an empty string is returned no matter the ``min_length`` probability: An optional tuples of length and probability mask for the text. The length is expected to be between the ``min_length`` and ``max_length``. Otherwise, a random integer between ``min_length`` and ``max_length`` is selected. For the probability mask, we expect a numpy array of length of the charset passed with ``dtype == np.float64``. The sum of the probability mask should be 1, otherwise an exception is raised. Returns: A sampled string from the space NzIOnly one of `mask` or `probability` can be provided, actual values: mask=z, probability=maskrr z2Expects all mask values to 0 or 1, actual values: probabilityzIExpects all probability mask values to be within 0 and 1, actual values: z=Expects the sum of the probability mask to be 1, actual sum: rzBTrying to sample with a minimum length > 0 (actual minimum length=zP) but the character mask is all zero meaning that no character could be sampled.)sizep) ValueError_validate_maskrint8all logical_orsumfloat64 logical_andisclose np_randomintegersrroneslen character_setchoicecharacter_listr!)r'r.r/length charlist_maskstrings r+sample Text.sampleNsh,   7[\`[aaop{o|}  $($7$7bggv$N !F(vvMM-1"4mq6HIXG WX 66-(1,$1BFF=4I$IM  $$($7$7RZZ% !F(vvNN=A#5}7IJo^_l^mnozzFF=)1kRSUSYSYZgShRijk&* )F >^^,,T__dooPQ>QRF  GGC(:(:$;Expects the Text sample length to be an integer, actual type: z*Expects the Text sample length be between z and z, actual length: zExpects the Text sample `z$` to be an np.ndarray, actual type: z ` to be type z, actual dtype: zexpects the Text sample `z` to be z, actual shape: ) isinstancerrr>rrrrrndarrayrshaper?)r'r.expected_dtype mask_typerBrCs r+r3Text._validate_masks  %   V 9+%Hd U V  IN U 9+%HT T U  $  ==V bjj _OPTU[P\~^ _6BRRbcpcvcvbwx y5 &&D&&'+ @*9+Xs4CUCU?V>X=YYijwj}j}i~ @$$rGc^[U[5(aBTR[U5s=::aTR::aO g[ U4SjU55$g)z?Return boolean specifying if x is a valid member of this space.c3@># UHoTR;v M g7f)N)r?).0cr's r+ Text.contains..s>Aq 2 22AsF)rJr&rr>rr5)r'xs` r+contains Text.containssB a  #a&;DOO;>A>>>rGcVSURSURSURS3$)z,Gives a string representation of this space.zText(z, z , charset=))rr charactersr's r+__repr__ Text.__repr__s-t'r$//):*T__DUUVWWrGc[U[5=(aY URUR:H=(a9 URUR:H=(a URUR:H$)z7Check whether ``other`` is equivalent to this instance.)rJr rrr?)r'others r+__eq__ Text.__eq__s\ ud # :5#3#33 :5#3#33 :""e&9&99  rGcUR$)z(Returns the character set for the space.)rr\s r+r?Text.character_set~~rGcUR$)z+Returns a tuple of characters in the space.)rr\s r+rAText.character_listsrGc URU$)zGReturns a unique index for each character in the space's character set.)r )r'chars r+character_indexText.character_indexs%%rGcUR$)z*Returns a string with all Text characters.)r#r\s r+r[Text.charactersrerGcg)zaThe flattened version is an integer array for each character, padded to the max character length.Tr\s r+is_np_flattenableText.is_np_flattenablesrG)r rrr#rr)rrrrrzfrozenset[str] | strrz int | np.random.Generator | None)NN)r.z1None | tuple[int | None, NDArray[np.int8] | None]r/z4None | tuple[int | None, NDArray[np.float64] | None]returnr&)r.@tuple[int | None, NDArray[np.int8] | NDArray[np.float64] | None]rMznp.dtyperNr&rrrs)rVrrrbool)rrr&)r`rrrrt)rrr)rrztuple[str, ...])rir&rrznp.int32)rrrt)__name__ __module__ __qualname____firstlineno____doc__r r%rEr3rWr]rapropertyr?rArjr[rp__static_attributes__ __classcell__)r*s@r+r r s &(415 */*/ */ & */ / */*/\EINREAELE EN %N %! % % J %DX &rGr )ry __future__rtypingrnumpyr numpy.typingrgymnasium.spaces.spacerrr __annotations__r&r rorGr+rs?@" ( )D  n O5:OrG