o 1 i4 @sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl ZddlZddlZddlmZddlmZddlmZddlmZddlmZd Zed efd d Zd ejdefddZ dedejfddZ!edej"j#defddZ$edej"j#defddZ%ededej"j#fddZ&ededej"j#fddZ'ede edeeeffd d!Z(ed"e e efdefd#d$Z)e %d*d&e ee fd'e*de e ee ffd(d)Z+dS)+N) OrderedDict)AnyDictOptionalSequenceTypeUnion) DeveloperAPI)NotSerializable)FlexDict)RepeatedSimplexZ__not_serializable__objcCst|tjr t|St|tjrt|St|tjrt|St|tjr(t |St|tj rC| }t |D] \}}t |||<q6|S|S)zConvert an object that is a numpy type to a python type. If the object is not a numpy type, it is returned unchanged. Args: obj: The object to convert. ) isinstancenpintegerintZfloatingfloatZbool_boolZstr_strndarraytolist enumerate"convert_numpy_to_python_primitives)rretivri/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/rllib/utils/serialization.pyrs     rarrayreturncCs.t}t||tt| dS)zPack numpy ndarray into Base64 encoded strings for serialization. This function uses numpy.save() instead of pickling to ensure compatibility. Args: array: numpy ndarray. Returns: b64 escaped string. ascii) ioBytesIOrsavebase64 b64encodezlibcompressgetvaluedecode)r bufrrr_serialize_ndarray/s r- b64_stringcCs tjttt|ddS)zUnpack b64 escaped string into numpy ndarray. This function assumes the unescaped bytes are of npy format. Args: b64_string: Base64 escaped string. Returns: numpy ndarray. T)Z allow_pickle)rloadr#r$r( decompressr& b64decode)r.rrr_deserialize_ndarray@s r2spacec Cs|durdSdtjjdtfdd}dtjjdtfdd}dtjjdtfdd }dtjjdtfd d }dtjjdtfd d }dtjjdtfdd}dtdtfdd}dt dtfdd}dt dtfdd} dddtfdd} t |tjjrw||St |tjjr||St |tjjr||St |tjjr||St |tjjr||St |tjjr||St |tjj r| |St |tr||St |t r||St |t r| |St dt|)zSerialize a gym Space into a JSON-serializable dict. Args: space: gym.spaces.Space Returns: Serialized JSON string. Nspr!cSs"dt|jt|j|j|jjdS)Nbox)r3lowhighshapedtype)r-r6r7_shaper9rr4rrr_box]s zgym_space_to_dict.._boxcSs,dt|jd}t|drt|j|d<|S)Ndiscreter3nstart)rr?hasattrr@)r4drrr _discretefs  z$gym_space_to_dict.._discretecSs d|jdS)N multi-binaryr>)r?r;rrr _multi_binarypsz(gym_space_to_dict.._multi_binarycSsdt|j|jjdS)Nmulti-discrete)r3nvecr9)r-rGr9rr;rrr_multi_discretevsz*gym_space_to_dict.._multi_discretecSsddd|jDdS)NtuplecSg|]}t|qSrgym_space_to_dict.0r4rrr z5gym_space_to_dict.._tuple..r3spacesrRr;rrr_tuple}sz!gym_space_to_dict.._tuplecSsddd|jDdS)NdictcSi|] \}}|t|qSrrKrNkr4rrr z4gym_space_to_dict.._dict..rQ)rRitemsr;rrr_dictsz gym_space_to_dict.._dictcSsd|j|j|jjdS)Nsimplex)r3r8 concentrationr9)r:r^r9rr;rrr_simplexs z#gym_space_to_dict.._simplexcSsdt|j|jdS)Nrepeated)r3 child_spacemax_len)rLrarbr;rrr _repeatedsz$gym_space_to_dict.._repeatedcSs(ddi}|jD] \}}t|||<q|S)Nr3 flex_dict)rRrL)r4rBrXsrrr _flex_dicts z%gym_space_to_dict.._flex_dictgym.spaces.TextcSsBt|dd}|durt|dd}|durtdd|j|j|dS)NZ character_setcharsetz9Text space must have a character_set or charset attributetext)r3 min_length max_lengthrh)getattr ValueErrorrjrk)r4rhrrr_texts  z gym_space_to_dict.._textz&Unknown space type for serialization, )gymrRBoxrDiscrete MultiBinary MultiDiscreteTuplerr r rTextrmtype) r3r<rCrErHrTr\r_rcrfrnrrrrLPsB      rLcCs(dt|i}d|jvrt|j|d<|SNr3original_space)rL__dict__ space_to_dictrx)r3rBrrrrzs  rzrBc s>|durdSdtfdddtdtjjffdd }dtdtjjffdd }dtdtjjffd d }dtdtjjffd d }dtdtjjfdd}dtdtjjfdd}dtdtffdd }dtdtfdd}dtdt fdd} dtddffdd } ||||||||| | d } |d} | | vrt d| | | |S)zDe-serialize a dict into gym Space. Args: str: serialized JSON str. Returns: De-serialized gym space. NrBcSs,|}|d=d|vrt|d|d<|S)z?Common updates to the dict before we use it to construct spacesr3r9)copyrr9rBrrrr__commons z%gym_space_from_dict..__commonr!cs>|}|t|dt|ddtjjdi|S)Nr6r7)r6r7r)r{updater2rorRrpr|r}rrr<s  z!gym_space_from_dict.._boxctjjdi|SNr)rorRrqrBrrrrCz&gym_space_from_dict.._discretecrr)rorRrrrrrrrErz*gym_space_from_dict.._multi_binarycs4|}|dt|ditjjdi|S)NrGr)r{r~r2rorRrsr|rrrrHs  z,gym_space_from_dict.._multi_discretecSs dd|dD}tjj|dS)NcSrJrgym_space_from_dictrMrrrrOrPz7gym_space_from_dict.._tuple..rRrS)rorRrtrBrRrrrrTsz#gym_space_from_dict.._tuplecSs(tdd|dD}tjj|dS)NcSrVrrrWrrrrY rZz6gym_space_from_dict.._dict..rRrS)rr[rorRrrrrrr\s z"gym_space_from_dict.._dictcstdi|Srr rrrrr_sz%gym_space_from_dict.._simplexcSst|d}t||ddS)Nrarb)rarb)rr )rBrarrrrcs z&gym_space_from_dict.._repeatedcSsdd|D}t|dS)NcSs"i|] \}}|dkr|t|qS)r3r)rNrXrerrrrYs"z;gym_space_from_dict.._flex_dict..rS)r[r rrrrrfs z'gym_space_from_dict.._flex_dictrgcrr)rorRrurrrrrnrz"gym_space_from_dict.._text) r5r=rDrFrIrUr]r`rdrir3z)Unknown space type for de-serialization, ) rrorRrprqrrrsrr r rm) rBr<rCrErHrTr\r_rcrfrnZ space_mapZ space_typerrrrs8      rcCsZt|d}d|vr+d|dvsJt|ddtr$t|d|_|St|d|_|Srw)rrrrxspace_from_dict)rBr3rrrr3s rargskwargsc Cs|D]"}z tt|Wqty$}z td|d|d}~ww|D]'\}}z tt|Wq)tyP}z td|d|d|d}~wwdS)a Check if parameters to a function are serializable by ray. Args: args: arguments to be checked. kwargs: keyword arguments to be checked. Raises: NoteSerializable if either args are kwargs are not serializable by ray. zVRLModule constructor arguments must be serializable. Found non-serializable argument: z . Original serialization error: Nz^RLModule constructor arguments must be serializable. Found non-serializable keyword argument: z = )raygetput TypeErrorr r[)rrargerXrrrr!check_if_args_kwargs_serializableBs: rtype_cCst|tr|S|jd|jS)zConverts a type into its full classpath ([module file] + "." + [class name]). Args: type_: The type to convert. Returns: The full classpath of the given type, e.g. "ray.rllib.algorithms.ppo.PPOConfig". .)rr __module__ __qualname__)rrrrserialize_typefs rFmoduleerrorc Cst|tr|St|trBz|dd\}}t|}t||WSttt t fyA}z|r6t d|d|WYd}~|Sd}~wwt d|d)aVResolves a class path to a class. If the given module is already a class, it is returned as is. If the given module is a string, it is imported and the class is returned. Args: module: The classpath (str) or type to resolve. error: Whether to throw a ValueError if `module` could not be resolved into a class. If False and `module` is not resolvable, returns None. Returns: The resolved class or `module` (if `error` is False and no resolution possible). Raises: ValueError: If `error` is True and `module` cannot be resolved. rz2Could not deserialize the given classpath `module=z` into a valid python class! Make sure you have all necessary pip packages installed and all custom modules are in your `PYTHONPATH` env variable.Nz `module` (z$ must be type or string (classpath)!) rrvrrsplit importlib import_modulerlModuleNotFoundError ImportErrorAttributeErrorrm)rr module_path class_namerrrrdeserialize_typexs&        r)F),r&rr#r( collectionsrtypingrrrrrrZ gymnasiumronumpyrrZray.rllib.utils.annotationsr Zray.rllib.utils.errorr Zray.rllib.utils.spaces.flexdictr Zray.rllib.utils.spaces.repeatedr Zray.rllib.utils.spaces.simplexrZNOT_SERIALIZABLErrrr-r2rRZSpacerLrzrrrrrrrrrrsP       wb #