o 2h@sdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z dZGdddejZd d Zd d Z dd dZ dddZddZdS)zJSON encoder and decoder that handles TensorShapes and tuples.cCs:tjrt|tjr|jdur|nd}d|dSt|S)zMEncodes objects for types that aren't handled by the default encoder.N TensorShape class_nameitems)tf available isinstancerrankas_list get_json_type)selfobjr r]/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/legacy/saving/json_utils.pydefaults zEncoder.defaultcstt|SN)superencode _encode_tuple)rr __class__rrrszEncoder.encode)__name__ __module__ __qualname____doc__rr __classcell__rrrrrsrcCsZt|trdtdd|DdSt|trdd|DSt|tr+dd|DS|S) N __tuple__cs|]}t|VqdSrr.0irrr $z _encode_tuple..rcSsg|]}t|qSrr#r$rrr 'sz!_encode_tuple..cSsi|] \}}|t|qSrr#)r%keyvaluerrr )sz!_encode_tuple..)r tuplelistdictr )xrrrr s   rcCstj|tdS)N object_hook)jsonloads_decode_helper) json_stringrrrdecode.sr7cCstj|tjtd||ddS)zFDecodes the JSON and deserializes any Keras objects found in the dict.T) deserializemodule_objectscustom_objectsr1)r3r4 functoolspartialr5)r6r9r:rrrdecode_and_deserialize2sr=Fc Cs`t|trd|vrtjra|ddkrt|dS|ddkr3ddlm}||dt |dS|dd kra|d }g}|d D]\}}| tj |tj |d qCtjjt ||d dS|ddkrrtdd|dDS|ddkrztS|rd|vrzd|vrtj|||dWStj|||dWStyY|Sw|ddkr|ddS|S)aJA decoding helper that is TF-object aware. Args: obj: A decoded dictionary that may represent an object. deserialize: Boolean. When True, deserializes any Keras objects found in `obj`. Defaults to `False`. module_objects: A dictionary of built-in objects to look the name up in. Generally, `module_objects` is provided by midlevel library implementers. custom_objects: A dictionary of custom objects to look the name up in. Generally, `custom_objects` is provided by the end user. Returns: The decoded object. r rr TypeSpecrtype_spec_registry type_spec serializedCompositeTensorspectensors)dtypeTexpand_compositesr!csr"r)r5r$rrrr'ir(z!_decode_helper.. __ellipsis____passive_serialization__module)r9r: __bytes__r+utf-8)r r/r r rtensorflow.python.frameworkr@lookup _deserializer5appendconstantdtypesas_dtypenestpack_sequence_asr-Ellipsisrdeserialize_keras_objectr ValueErrorr) rr8r9r:r@rDrErFtensorrrrr5AsV           r5c Cst|drt|}d|d<|St|jtjkr&t|tjr"| S| St |r-|jSt|jtjkr8|jSt j rFt|t jjjrF|jSt j rSt|t jrS|St j r_t|t jr_|jSt|tjjrjt|S|turrddiSt j rt|t jrddlm}z|t|}d||d WSt yt d |d t|d wt j rt|t j!j"rt #|}g}t j$j%|dd D]}|&|j'j|( fqdt)||dSt|t*j+r|jSt|t,rd|-ddSt.d |dt|d)zSerializes any object to a JSON-serializable structure. Args: obj: the object to serialize Returns: JSON-serializable structure representing `obj`. Raises: TypeError: if `obj` cannot be serialized. get_configTrJr rIrr?r>)r rArBzUnable to serialize z% to JSON, because the TypeSpec class z has not been registered.rGrC)r rDrErLrM)r r+z to JSON. Unrecognized type .)/hasattrrserialize_keras_objecttypernprr ndarraytolistitemcallabler r compatv1 Dimensionr+rrDTypename collectionsabcMappingr/rWr>rNr@get_name _serializerY __internal__rCtype_spec_from_valuerUflattenrQrFnumpyrenumEnumbytesr7 TypeError)rrBr@type_spec_namerDrErZrrrrsj        r)NN)FNN)rrjrsr;r3rrr`keras.src.legacy.savingrkeras.src.savingrkeras.src.utils.module_utilsrr r JSONEncoderrrr7r=r5rrrrrs$     D